def test_mat_rows_are_vecs(self): mat = Mat( [1, 2, 3], [4, 5, 6], [7, 8, 9], ) self.assertEqual(mat.row(0), Vec(1, 2, 3))
def test_normalize_row(self): mat = Mat( [1, 2, 3], [0, 2, 0], [1, 0, 0], ) correct = Mat( [1, 2, 3], [0, 1, 0], [1, 0, 0], ) vec = mat.row(1) vec.normalize_in_place() self.assertEqual(mat, correct)