(self)
| 71 | class TestModule(unittest.TestCase): |
| 72 | |
| 73 | def test_module(self): |
| 74 | m = Module3() |
| 75 | print(m) |
| 76 | m.forward() |
| 77 | |
| 78 | self.assertEqual(4, len(list(m.named_modules()))) |
| 79 | self.assertEqual(5, len(list(m.named_network_outputs()))) |
| 80 | self.assertEqual( |
| 81 | [("", m), ("m1", m.m1), ("m1.m1", m.m1.m1), ("m1.m2", m.m1.m2)], |
| 82 | list(m.named_modules()), |
| 83 | ) |
| 84 | self.assertEqual( |
| 85 | [("", m, None), ("m1", m.m1, m), ("m1.m1", m.m1.m1, m.m1), |
| 86 | ("m1.m2", m.m1.m2, m.m1)], |
| 87 | list(m.named_modules_with_parent()), |
| 88 | ) |
| 89 | |
| 90 | def test_module_list(self): |
| 91 | m = Module4() |
nothing calls this directly
no test coverage detected