(self)
| 305 | |
| 306 | class FanAxesTest(TestCase): |
| 307 | def test_normalize(self): |
| 308 | fan_axes = FanAxes(in_axis=0, out_axis=(1,), batch_axis=(2, 1)) |
| 309 | fan_axes = fan_axes.canonicalize() |
| 310 | self.assertEqual(fan_axes.in_axis, (0,)) |
| 311 | self.assertEqual(fan_axes.out_axis, (1,)) |
| 312 | self.assertEqual(fan_axes.batch_axis, (1, 2)) |
| 313 | |
| 314 | def test_eq(self): |
| 315 | self.assertEqual( |
nothing calls this directly
no test coverage detected