def test_1(self): A = Group(1, "V = B['V']") A[...] = 1 B = Group(1, "V = A['V']") B[...] = 2 A.setup() B.setup() A.evaluate(dt=1, update=False) B.evaluate(dt=1, update=False) A.update() B.update() assert A['V'][0] == 2 and B['V'][0] == 1
class GroupMask(unittest.TestCase): def setUp(self): self.Z = Group((3,3)) def test_mask_true(self): self.Z.mask = True self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0,np.ones((3,3))) def test_mask_1(self): self.Z.mask = 1 self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0,np.ones((3,3))) def test_mask_false(self): self.Z.mask = False self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0,np.zeros((3,3))) def test_mask_0(self): self.Z.mask = 0 self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0,np.zeros((3,3))) def test_mask_shape_0(self): self.Z.mask = np.zeros((3,3)) self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0,np.zeros((3,3))) def test_mask_shape_1(self): self.Z.mask = np.ones((3,3)) self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0,np.ones((3,3))) def test_mask_shape_2(self): mask = np.ones((3,3)) mask[1,1] = 0 self.Z.mask = mask self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0,mask)
class GroupMask(unittest.TestCase): def setUp(self): self.Z = Group((3, 3)) def test_mask_true(self): self.Z.mask = True self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0, np.ones((3, 3))) def test_mask_1(self): self.Z.mask = 1 self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0, np.ones((3, 3))) def test_mask_false(self): self.Z.mask = False self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0, np.zeros((3, 3))) def test_mask_0(self): self.Z.mask = 0 self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0, np.zeros((3, 3))) def test_mask_shape_0(self): self.Z.mask = np.zeros((3, 3)) self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0, np.zeros((3, 3))) def test_mask_shape_1(self): self.Z.mask = np.ones((3, 3)) self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0, np.ones((3, 3))) def test_mask_shape_2(self): mask = np.ones((3, 3)) mask[1, 1] = 0 self.Z.mask = mask self.Z[...] = 1 self.Z.setup() assert np_equal(self.Z.f0, mask)