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")
