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