def test_fill(self): m1 = VectorMatrix(Shape(10, 10, 10)); m1.fill((1.0, 2.0, 3.0)); self.assertEqual(m1.get(4, 4, 4), (1.0, 2.0, 3.0)) self.assertEqual(m1.uniform_value, (1.0, 2.0, 3.0)) self.assertTrue(m1.isUniform()) m1.set(4, 4, 4, (3.0, 2.0, 1.0)) self.assertEqual(m1.get(4, 4, 4), (3.0, 2.0, 1.0)) self.assertEqual(m1.get(0, 1, 2), (1.0, 2.0, 3.0)) self.assertFalse(m1.isUniform())
def test_add_1(self): A = VectorMatrix(Shape(100,100,10)) B = VectorMatrix(Shape(100,100,10)) for idx in range(A.size()): A.set(idx, (1*idx, 2*idx, 3*idx)) B.set(idx, (4*idx, 5*idx, 6*idx)) A.add(B, 2.0) for idx in range(A.size()): self.assertEqual(A.get(idx), (9*idx, 12*idx, 15*idx))
def test_fill(self): A = VectorMatrix(Shape(100,100,10)) A.fill((1.0, 2.0, 3.0)) for idx in range(A.size()): self.assertTrue(A.get(idx) == (1.0, 2.0, 3.0))