def test_solve(self): """ test solve """ read = StringIO("10089:\n662823\n1693997\n1985602\n2345955\n1205751\n") word = StringIO() Netflix.netflix_solve(read, word) self.assertEqual(word.getvalue(), "10089:\n2.4\n4.4\n2.8\n3.0\n5.0\nRMSE: 0.59\n")
def test_solve_3(self): r = StringIO("0:\n2\n") w = StringIO() Netflix.netflix_solve(r, w, False) self.assertEqual(w.getvalue(), "0:\n2.0\nRMSE: 2.0\n")
def test_solve_1(self): r = StringIO("0:\n0\n") w = StringIO() Netflix.netflix_solve(r, w, False) self.assertEqual(w.getvalue(), "0:\n1.3\nRMSE: 1.7\n")