def test_netflixsolve3(self):
r = StringIO(
"1000:\n2326571\n977808\n1010534\n1861759\n79755\n98259\n1960212\n97460\n2623506\n2409123\n1959111\n809597\n2251189\n537705\n929584\n506737\n708895\n1900790\n2553920\n1196779\n2411446\n1002296\n1580442\n100291\n433455\n2368043\n906984\n10008:\n1813636\n2048630\n930946\n1492860\n1687570\n1122917\n1885441\n10009:\n1927533\n1789102\n2263612\n964421\n701514\n2120902\n"
)
w = StringIO()
RMSE = netflixSolve(w, r)
self.assertTrue(RMSE < 1)
def test_netflixsolve1(self):
r = StringIO(
"1:\n30878\n2647871\n1283744\n2488120\n317050\n1904905\n1989766\n14756\n1027056\n1149588\n1394012\n1406595\n2529547\n1682104\n2625019\n2603381\n1774623\n470861\n712610\n1772839\n1059319\n2380848\n548064\n"
)
w = StringIO()
RMSE = netflixSolve(w, r)
self.assertTrue(RMSE < 1)
def test_netflixsolve2(self):
r = StringIO("10:\n1952305\n1531863")
w = StringIO()
RMSE = netflixSolve(w, r)
self.assertTrue(RMSE < 1)
import sys from Netflix import netflixSolve netflixSolve(sys.stdin, sys.stdout)
def test_solve (self):
r = StringIO("108:\n1263043\n")
w = StringIO()
netflixSolve(r,w)
self.assertEqual(w.getvalue(), "108:\n4.3\nRMSE: 0.67")
#!/usr/bin/env python3
# -------
# imports
# -------
import sys
from Netflix import netflixSolve
# ----
# main
# ----
if __name__ == '__main__':
netflixSolve(sys.stdout, sys.stdin)
