|
24 | 24 | )
|
25 | 25 |
|
26 | 26 |
|
27 |
| -def test_component(name: str, role_name: str = "worker") -> AppDef: |
| 27 | +def _test_component(name: str, role_name: str = "worker") -> AppDef: |
28 | 28 | """
|
29 | 29 | Test component
|
30 | 30 |
|
@@ -94,10 +94,10 @@ def test_get_entrypoints_components(self) -> None:
|
94 | 94 | with patch("torchx.specs.finder.entrypoints") as entrypoints_mock:
|
95 | 95 | entrypoints_mock.load_group.return_value = test_torchx_group
|
96 | 96 | components = _load_components()
|
97 |
| - foobar_component = components["foobar.finder_test.test_component"] |
98 |
| - self.assertEqual(test_component, foobar_component.fn) |
99 |
| - self.assertEqual("test_component", foobar_component.fn_name) |
100 |
| - self.assertEqual("foobar.finder_test.test_component", foobar_component.name) |
| 97 | + foobar_component = components["foobar.finder_test._test_component"] |
| 98 | + self.assertEqual(_test_component, foobar_component.fn) |
| 99 | + self.assertEqual("_test_component", foobar_component.fn_name) |
| 100 | + self.assertEqual("foobar.finder_test._test_component", foobar_component.name) |
101 | 101 | self.assertEqual("Test component", foobar_component.description)
|
102 | 102 |
|
103 | 103 | def test_get_base_module_name(self) -> None:
|
@@ -142,20 +142,20 @@ def current_file_path() -> str:
|
142 | 142 | class CustomComponentsFinderTest(unittest.TestCase):
|
143 | 143 | def test_find_components(self) -> None:
|
144 | 144 | components = CustomComponentsFinder(
|
145 |
| - current_file_path(), "test_component" |
| 145 | + current_file_path(), "_test_component" |
146 | 146 | ).find()
|
147 | 147 | self.assertEqual(1, len(components))
|
148 | 148 | component = components[0]
|
149 |
| - self.assertEqual(f"{current_file_path()}:test_component", component.name) |
| 149 | + self.assertEqual(f"{current_file_path()}:_test_component", component.name) |
150 | 150 | self.assertEqual("Test component", component.description)
|
151 |
| - self.assertEqual("test_component", component.fn_name) |
| 151 | + self.assertEqual("_test_component", component.fn_name) |
152 | 152 | self.assertListEqual([], component.validation_errors)
|
153 | 153 |
|
154 | 154 | def test_get_component(self) -> None:
|
155 |
| - component = get_component(f"{current_file_path()}:test_component") |
156 |
| - self.assertEqual(f"{current_file_path()}:test_component", component.name) |
| 155 | + component = get_component(f"{current_file_path()}:_test_component") |
| 156 | + self.assertEqual(f"{current_file_path()}:_test_component", component.name) |
157 | 157 | self.assertEqual("Test component", component.description)
|
158 |
| - self.assertEqual("test_component", component.fn_name) |
| 158 | + self.assertEqual("_test_component", component.fn_name) |
159 | 159 | self.assertListEqual([], component.validation_errors)
|
160 | 160 |
|
161 | 161 | def test_get_components(self) -> None:
|
|
0 commit comments