(self)
| 312 | self.assertEqual(fan_axes.batch_axis, (1, 2)) |
| 313 | |
| 314 | def test_eq(self): |
| 315 | self.assertEqual( |
| 316 | FanAxes(in_axis=(0,), out_axis=1), |
| 317 | FanAxes(in_axis=0, out_axis=(1,)), |
| 318 | ) |
| 319 | self.assertNotEqual( |
| 320 | FanAxes(in_axis=(0,), out_axis=1), |
| 321 | 42, |
| 322 | ) |
| 323 | self.assertNotEqual( |
| 324 | FanAxes(in_axis=(0,), out_axis=1), |
| 325 | FanAxes(in_axis=0, out_axis=(2,)), |
| 326 | ) |
| 327 | |
| 328 | # pytype: disable=attribute-error |
| 329 | def test_prepend_axis(self): |