def test_mat_from_vecs(self): v1 = Vec(1, 2, 3) v2 = Vec(4, 5, 6) mat = Mat(v1, v2) correct = Mat([ [1, 2, 3], [4, 5, 6] ]) self.assertEqual(mat, correct) v1.z = 39 v2.y = 25 correct = Mat([ [1, 2, 39], [4, 25, 6] ]) self.assertEqual(mat, correct)
def test_write_access(self): vec = Vec(1, 2, 3) vec.y = 49 self.assertEqual(vec[1], 49) self.assertEqual(vec.y, 49)