class UtilsTest(unittest.TestCase): def setUp(self): self.s = Solver() self.x = np.array([ [[P11], [PN1]] ]) self.xx = np.append(self.x, [[[P12], [PN2]]], axis=0) self.xxx = np.append(self.xx, [[[P13], [PN3]]], axis=0) self.theta = np.array([ [B0], [B1] ]) def test_solve(self): for item in [1, 2, 4]: self.assertEqual( self.s.solve(np.repeat(self.theta, item), self.x)[0, 0], 100 / item ) def test_precision_solve(self): for item in [1, 2, 4]: l = len(self.s.solve(np.repeat(self.theta, item), self.xx)) self.assertEqual( self.s.solve(np.repeat(self.theta, item), self.xx)[0, 0], 100 / item ) self.assertEqual( self.s.solve(np.repeat(self.theta, item), self.xx)[l / 2, 0], 200 / item )
def setUp(self): self.s = Solver() self.x = np.array([ [[P11], [PN1]] ]) self.xx = np.append(self.x, [[[P12], [PN2]]], axis=0) self.xxx = np.append(self.xx, [[[P13], [PN3]]], axis=0) self.theta = np.array([ [B0], [B1] ])