def testSort2(self): m = sympy.Matrix([[0, 0, 0], [0, 1, 1], [1, 0, 0]]) exp = sympy.Matrix([[1, 0, 0], [0, 1, 1], [0, 0, 0]]) self.assertEqual(exp, matrix.sort(m))
def testSort1(self): m = sympy.Matrix([[3, 1, 2], [4, 1, 2], [2, 1, 2]]) exp = sympy.Matrix([[4, 1, 2], [3, 1, 2], [2, 1, 2]]) self.assertEqual(exp, matrix.sort(m))