def test_average(self): A = VectorMatrix(Shape(10,10,10)) for idx in range(A.size()): A.set(idx, (idx, 2*idx, 3*idx)) self.assertTrue(A.average() == (999.0/2,2*999.0/2,3*999.0/2))