def test_solve_2(self): """ test solve """ string_in = StringIO("10006:\n1093333\n1982605\n1534853\n") writer = StringIO() netflix_solve(string_in, writer) self.assertEqual(writer.getvalue(), "10006:\n3.7\n3.1\n3.9\nRMSE: 1.26\n")
def test_solve_3(self): """ test solve """ string_in = StringIO("10010:\n1462925\n52050\n650466\n") writer = StringIO() netflix_solve(string_in, writer) self.assertEqual(writer.getvalue(), "10010:\n2.2\n2.0\n2.0\nRMSE: 1.33\n")
def test_solve_2(self): """ test solves input 2""" inp = StringIO("1:\n30878\n2043:\n462685\n") writer = StringIO() netflix_solve(inp, writer) self.assertEqual( writer.getvalue(), "1:\n3.7\n2043:\n4.0\nRMSE: 0.18432622758011358\n")
def test_solve_3(self): """ test solves input 3""" inp = StringIO("2043:\n2312054\n1:\n30878\n") writer = StringIO() netflix_solve(inp, writer) self.assertEqual( writer.getvalue(), "2043:\n4.6\n1:\n3.7\nRMSE: 2.5588590159754325\n")
def test_solve_1(self): """ test solves input 1""" inp = StringIO("2043:\n1417435\n2312054\n462685\n") writer = StringIO() netflix_solve(inp, writer) self.assertEqual( writer.getvalue(), "2043:\n3.7\n4.6\n4.0\nRMSE: 2.1283858730975753\n")
def test_netflix_solve_3 (self) : re = StringIO.StringIO("COSTUMERS_RATE\n1 3.4\n2 4.4\n3 2.4\n5 1.2\n4 2.2\nMOVIE_RATES\n2 2.5\n1 3.5") getCatch(re) r = StringIO.StringIO("1:\n2:") w = StringIO.StringIO() netflix_solve(r, w) self.assert_(w.getvalue() == "1:\n2:\n")
def test_solve_3(self): """Test 3""" str_in = StringIO("1:\n30878\n2647871\n1283744") str_out = StringIO() netflix_solve(str_in, str_out) self.assertEqual( str_out.getvalue(), "1:\n3.7\n3.3\n3.6\nRMSE: 0.57")
def test_solve_1(self): """Test netflix_solve""" reader = StringIO("1123:\n448549\n2444222\n1522889\n") writer = StringIO() netflix_solve(reader, writer) self.assertEqual(writer.getvalue(), "1123:\n3.3\n2.9\n4.1\nRMSE: 0.4148335185443161\n")
def test_solve_1(self): """Test netflix_solve""" reader = StringIO("1:\n1\n2\n3\n4\n5\n6\n2:\n1\n2\n3\n4\n") writer = StringIO() true_dict = { ("1", "1"): 5, ("1", "2"): 1, ("1", "3"): 2, ("1", "4"): 4, ("1", "5"): 1, ("1", "6"): 2, ("2", "1"): 5, ("2", "2"): 4, ("2", "3"): 5, ("2", "4"): 5 } movie_avg_dict = {1: 3.2, 2: 4.1} cust_avg_dev_dict = {1: 0.7, 2: 0.3, 3: 0.1, 4: 0.1, 5: 0.5, 6: 0.8} netflix_solve(reader, writer, true_dict, movie_avg_dict, cust_avg_dev_dict) self.assertEqual( writer.getvalue(), "1:\n3.9\n3.5\n3.3\n3.3\n3.7\n4.0\n2:\n4.8\n4.3\n4.1\n4.1\n1.5")
def test_netflix_solve_6(self): """ Test 6 to see if solving correctly """ read = StringIO("1:\n1\n2\n3\n4\n2:\n5") write = StringIO() actual_ratings = {1: {1: 4, 2: 5, 3: 5, 4: 5}, 2: {5: 5}} movie_cache = {'1': 4.5, '2': 2.1525} movie_year_cache = {1: 1988, 2: 1975} cust_year_cache = { 1: { 1988: 2.5 }, 2: { 1988: 2.0 }, 3: { 1988: 4.3 }, 4: { 1988: 4.9 }, 5: { 1975: 3.9 } } netflix_solve(read, write, actual_ratings, movie_cache, movie_year_cache, cust_year_cache) self.assertEqual(write.getvalue(), "1:\n3.3\n2.8\n5.0\n5.0\n2:\n2.4\nRMSE: 1.57\n")
def test_solve_3(self): """ test solves input 3""" inp = StringIO("2043:\n2312054\n1:\n30878\n") writer = StringIO() netflix_solve(inp, writer) self.assertEqual( writer.getvalue(), "2043:\n4.6\n1:\n3.7\nRMSE: 2.5264965005457753\n")
def test_solve_1(self): """ test solves input 1""" inp = StringIO("2043:\n1417435\n2312054\n462685\n") writer = StringIO() netflix_solve(inp, writer) self.assertEqual( writer.getvalue(), "2043:\n3.6\n4.6\n3.9\nRMSE: 2.087128003334406\n")
def test_solve_2(self): """ test solves input 2""" inp = StringIO("1:\n30878\n2043:\n462685\n") writer = StringIO() netflix_solve(inp, writer) self.assertEqual( writer.getvalue(), "1:\n3.7\n2043:\n3.9\nRMSE: 0.2180552033726668\n")
def test_netflix_solve_4(self): """ test_netflix_solve_4 """ reader = StringIO("2043:\n1417435\n1:\n30878") writer = StringIO() netflix_solve(reader, writer) self.assertEqual(writer.getvalue(), "1:\n30878 3.7\n2043:\n1417435 3.6\nRMSE: 0.47")
def test_netflix_solve_3(self): """ Test 3 to see if solving correctly """ read = StringIO("1:\n1\n2\n3\n4\n2:\n5") write = StringIO() actual_ratings = {1: {1: 2, 2: 3, 3: 4, 4: 3}, 2: {5: 1}} movie_cache = {'1': 4.5, '2': 2.1525} movie_year_cache = {1: 1988, 2: 1975} cust_year_cache = { 1: { 1988: 1 }, 2: { 1988: 4.987 }, 3: { 1988: 5 }, 4: { 1988: 3 }, 5: { 1975: 3 } } netflix_solve(read, write, actual_ratings, movie_cache, movie_year_cache, cust_year_cache) self.assertEqual(write.getvalue(), "1:\n1.8\n5.0\n5.0\n3.8\n2:\n1.5\nRMSE: 1.08\n")
def test_solve(self): r = StringIO("1:\n30878\n2647871\n1283744\n") w = StringIO() netflix_solve(r, w) self.assertEqual( w.getvalue(), "1:\n3.7\n3.3\n3.6\nRMSE: 0.53 (two decimal places)\n")
def test_solve_2(self): """Sample run throughs""" str_in = StringIO("10:\n1952305\n1531863\n1000:\n2326571\n977808") str_out = StringIO() netflix_solve(str_in, str_out) self.assertEqual( str_out.getvalue(), "10:\n2.9\n2.6\n1000:\n3.2\n2.9\nRMSE: 0.22")
def test_netflix_solve_2(self): """ test_netflix_solve_2 """ reader = StringIO("2043:\n1417435\n2312054") writer = StringIO() netflix_solve(reader, writer) self.assertEqual(writer.getvalue(), "2043:\n1417435 3.6\n2312054 4.3\nRMSE: 2.37")
def test_netflix_solve_5(self): """ Test 5 to see if solving correctly """ read = StringIO("1:\n1\n2\n3\n4\n2:\n5") write = StringIO() actual_ratings = {1: {1: 4, 2: 5, 3: 5, 4: 5}, 2: {5: 5}} movie_cache = {'1': 3.7, '2': 1.75} movie_year_cache = {1: 1988, 2: 1975} cust_year_cache = { 1: { 1988: 1.5 }, 2: { 1988: 2.5 }, 3: { 1988: 4.6 }, 4: { 1988: 2.9 }, 5: { 1975: 3.6 } } netflix_solve(read, write, actual_ratings, movie_cache, movie_year_cache, cust_year_cache) self.assertEqual(write.getvalue(), "1:\n1.5\n2.5\n4.6\n2.9\n2:\n1.6\nRMSE: 2.38\n")
def test_solve_2(self): w = StringIO() netflix_solve(self.input_small2, w) self.assertEqual( w.getvalue(), "9996:\n2.43\n2.77\n3.67\n3.43\n3.58\n" "2.96\n4.37\n3.45\n2.94\n3.42\n3.70\n" "2.60\n3.57\n3.37\n2.56\n3.66\n2.91\n" "4.23\n3.81\nRMSE: 1.45")
def test_solve_4(self): """solve test 4""" read = 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\n123456\n' ) write = StringIO() netflix_solve(read, write) self.assertNotEqual(write.getvalue(), '')
def test_solve_2(self): """ Solves for a single movie """ read = StringIO("10:\n1952305\n1531863\n") write = StringIO() netflix_solve(read, write) self.assertEqual(write.getvalue(), "10:\n3.2\n3.1\nRMSE: 0.15\n")
def test_solve_1(self): """ test solve """ reader = StringIO("1:\n30878\n") writer = StringIO() netflix_solve(reader, writer) self.assertEqual(writer.getvalue(), "1:\n3.7\nRMSE: 0.3\n")
def test_solve_2(self): r = StringIO("10:\n1952305\n1531863\n1000:\n2326571\n977808\n") w = StringIO() netflix_solve(r, w) self.assertEqual( w.getvalue(), "10:\n2.9\n2.7\n1000:\n3.2\n2.9\nRMSE: 0.44 (two decimal places)\n" )
def test_solve_3(self): """ Solves for a single movie """ read = StringIO("10000:\n200206\n523108\n") write = StringIO() netflix_solve(read, write) self.assertEqual(write.getvalue(), "10000:\n3.6\n3.5\nRMSE: 1.05\n")
def test_solve1(self): """ Test that output is in string format """ reader = StringIO("1:\n30878\n2647871\n1283744\n2488120") writer = StringIO() netflix_solve(reader, writer) self.assertIsInstance(writer.getvalue(), str)
def test_netflix_solve_3(self): """ test_netflix_solve_3 """ reader = StringIO("2043:\n1417435\n2312054\n462685") writer = StringIO() netflix_solve(reader, writer) self.assertEqual( writer.getvalue(), "2043:\n1417435 3.6\n2312054 4.3\n462685 3.8\nRMSE: 1.94")
def test_solve_1(self): """ Solves for a single movie """ read = 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") write = StringIO() netflix_solve(read, write) self.assertEqual(write.getvalue(), "1:\n3.6\n3.4\n3.6\n4.2\n3.6\n3.7\n3.5\n3.6\n3.8\n3.6\n3.3\n3.6\n3.8\n3.8\n3.2\n3.8\n3.6\n4.1\n3.9\n3.9\n3.3\n4.3\n3.6\nRMSE: 0.82\n")
def test_solve_1(self): """Example""" str_in = StringIO( "13:\n615010\n1860468\n2131832\n") str_out = StringIO() netflix_solve(str_in, str_out) self.assertEqual( str_out.getvalue(), "13:\n4.9\n4.7\n5.0\nRMSE: 0.22")
def test_solve_2(self): ''' Testing the netflix_solve() function ''' read_io = StringIO("10000:\n200206") write_io = StringIO() netflix_solve(read_io, write_io) self.assertEqual(write_io.getvalue(), "10000:\n5.0\nRMSE: 0.0")
def test_solve_3(self): ''' Testing the netflix_solve() function ''' read_io = StringIO("10:\n1531863") write_io = StringIO() netflix_solve(read_io, write_io) self.assertEqual(write_io.getvalue(), "10:\n3.0\nRMSE: 0.02")
def test_netflix_solve_3 (self) : r = StringIO('2043:\n1417435\n2312054\n462685\n2849:\n1198093\n1129293\n1417435\n1894792\n') w = StringIO() netflix_solve(r,w) val = findall(r'\d\.\d\d+',w.getvalue()) # print(val) self.assertEqual(len(val),1) self.assertTrue(float(val[0]) < 1.75)
def test_solve(self): ''' Testing the netflix_solve() function ''' read_io = StringIO("1:\n317050\n1772839") write_io = StringIO() netflix_solve(read_io, write_io) self.assertEqual(write_io.getvalue(), "1:\n4.9\n5.0\nRMSE: 0.05")
def test_solve_1(self): """ Tests solve using test version """ reader = StringIO("8191:\n2104692\n") writer = StringIO() netflix_solve(reader, writer) self.assertEqual(writer.getvalue(), "8191:\n4.3\nRMSE: 0.31\n")
def test_solve1(self): """ testing netflix_solve """ read = StringIO("1:\n30878\n2647871\n1283744\n") write = StringIO() netflix_solve(read, write) self.assertEqual(write.getvalue(), "1:\n3.7\n3.0\n3.9\nRMSE: 0.79\n")
def test_solve_1 (self) : r = StringIO("1:\n30878\n2647871\n1283744") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "1:\n3.7\n3.3\n3.8\nRMSE: 0.63")
def test_solve_1(self): r = StringIO("1:\n548064\n") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "1:\n3.6\n")
def test_solve_2 (self) : r = StringIO("6683:\n2505263\n1237027\n") w = StringIO() netflix_solve(r,w) self.assertEqual (w.getvalue(), "6683:\n2.8\n1.9\nRMSE: 1.49")
#!/usr/bin/env python3 import sys, os from Netflix import netflix_solve # ---- # main # ---- if __name__ == "__main__": netflix_solve(sys.stdin, sys.stdout)
def test_solve_2 (self) : r = StringIO("1:\n30878\n2647871\n1283744\n") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "3.7\n3.5\n3.6\n")
def test_solve_1 (self) : r = StringIO("12126:\n1120091\n21560\n1684241\n") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "12126:\n4.1\n3.2\n3.9\nRMSE: 1.28")
def test_solve_2 (self) : r = StringIO("2043:\n1417435\n2312054\n2043:\n1417435\n2312054") #two sets w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "2043:\n3.6\n4.6\n2043:\n3.6\n4.6\nRMSE:2.5807\n")
def test_netflix_solve_2 (self) : r = StringIO('2043:\n1417435\n2312054\n462685\n2849:\n1198093\n1129293\n1417435\n1894792\n') w = StringIO() netflix_solve(r,w) self.assertNotEqual(w.getvalue(),'') self.assertTrue('RMSE:' in w.getvalue())
def test_solve1 (self) : r = io.StringIO("1:\n14756\n") w = io.StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "1:\n3.7\nRMSE: 0.30654350027973454\n")
def test_solve_3 (self) : r = StringIO("430:\n2573892\n1843756\n2540558\n68149\n2143918\n164344\n") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "430:\n4.9\n3.7\n4.4\n4.1\n3.1\n3.7\nRMSE: 2.08")
def test_solve_2 (self) : r = StringIO("4327:\n2604511\n1862650\n1456536\n") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "4327:\n3.4\n3.0\n3.2\nRMSE: 0.83")
def test_solve_2 (self) : r = StringIO("10000:\n2311278\n928262") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "10000:\n3.5\n3.1\nRMSE: 1.23")
def test_solve_4 (self): r = StringIO("") w = StringIO() netflix_solve(r,w) self.assertEqual(w.getvalue(),"RMSE:0.0\n")
def test_solve_3 (self) : r = StringIO("16575:\n30878\n2647871\n1283744\n2311278") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "16575:\n3.7\n3.7\n3.8\n3.9\nRMSE: 1.07")
def test_solve_3 (self) : r = StringIO("668:\n1830\n277783\n1300110") w = StringIO() netflix_solve(r,w) self.assertEqual (w.getvalue(), "668:\n4.4\n4.1\n4.0\nRMSE: 0.68")
def test_solve_3 (self) : # r = StringIO("2043:\n1417435\n2312054\n2043:\n1417435\n2312054\n2043:") #two sets and an incomplete w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "2043:\n3.6\n4.6\n2043:\n3.6\n4.6\n2043:\nRMSE:2.5807\n")
# --------------------------- # CS 373 Project 3 # Anh-Khoi Dang - add562 # --------------------------- # ------- # imports # ------- import sys, time from Netflix import netflix_read, netflix_eval, netflix_solve # ---- # main # ---- #s = time.clock() if len(sys.argv) != 4 : exit("Usage: python RunNetflix.py <movie titles file> <training set directory> <probe file>") else : movieTitlesFile = sys.argv[1] trainingSetDir = sys.argv[2] probeFile = sys.argv[3] netflix_solve(sys.stdout, trainingSetDir, probeFile) #e = time.clock() #print (e - s), "seconds"
def test_solve_1 (self) : r = StringIO("1:\n30878\n") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "3.7\n")
def test_solve_2 (self) : input_string = StringIO() output_string = StringIO() netflix_solve(input_string, output_string) self.assertEqual("RMSE: 0.00\n", output_string.getvalue())
def test_solve_3 (self) : r = StringIO("1:\n30878\n10:\n1952305\n1531863\n") w = StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "3.7\n3.3\n3.2\n")
def test_solve2 (self) : r = io.StringIO("1:\n712610\n1772839\n1059319\n2380848\n548064\n") w = io.StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "1:\n4.1\n4.2\n3.1\n5.0\n3.5\nRMSE: 0.7601868708858408\n")
def test_solve_3 (self) : input_string = StringIO("10:\n1952305\n1:\n30878\n") output_string = StringIO() netflix_solve(input_string, output_string) self.assertEqual("1:\n3.7\n10:\n3.3\nRMSE: 0.00\n", output_string.getvalue())
def test_solve_1 (self) : r = StringIO("1:\n30878\n317050\n") w = StringIO() netflix_solve(r,w) self.assertEqual (w.getvalue(), "1:\n3.8\n3.9\nRMSE: 0.79")
def test_solve4 (self) : r = io.StringIO("3676:\n1982226\n1495499\n1669842\n2639376\n1982597\n1426939\n") w = io.StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "3676:\n3.7\n3.6\n4.5\n2.9\n4.2\n4.2\nRMSE: 0.6597351087461958\n")
def test_solve3 (self) : r = io.StringIO("10007:\n1204847\n2160216\n248206\n") w = io.StringIO() netflix_solve(r, w) self.assertEqual(w.getvalue(), "10007:\n2.0\n2.7\n2.2\nRMSE: 1.6254316644013882\n")