|
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