def test_simple_case(self):
totals, m, e = self.setUp()
r_hat = column_scaling(m, totals, e)
self.assertTrue((r_hat == [[5, 0], [0, 10]]).all())
def test_type(self):
totals, m, e = self.setUp()
result = column_scaling(m, totals, e)
self.assertIs(type(result), numpy.matrix)