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
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)
