def test_flat(self): shape = (3, 3) t = np.arange(9).reshape(shape) z = Bicomplex(t, 2 * t) t = z.flat(1) assert t == Bicomplex(1, 2)
def test_flat(self): shape = (3, 3) t = np.arange(9).reshape(shape) z = Bicomplex(t, 2 * t) t = z.flat(1) self.assertTrue(t == Bicomplex(1, 2))