def test_print_2(self): """ tests printing movie_id """ rating = 2043 writer = StringIO() netflix_print(writer, rating) self.assertEqual(writer.getvalue(), "2043:\n")
def test_print_1(self): """ Prints a movie id """ temp1 = StringIO() netflix_print(1, 1, temp1) self.assertEqual(temp1.getvalue(), "1:\n")
def test_print_1(self): """ tests printing rating """ rating = 1.0 writer = StringIO() netflix_print(writer, rating) self.assertEqual(writer.getvalue(), "1.0\n")
def test_netflix_print_3(self): """ netflix_print_3 """ input_dic = {1: {1001, 1002, 1003, 1004, 1005}} store_rating = {1: {3.5, 3.6}} writer = StringIO() netflix_print(writer, input_dic, store_rating) self.assertEqual(writer.getvalue(), "1:\n1001 3.5\n1002 3.6\n")
def test_print_3(self): """ Prints a customer rating """ temp1 = StringIO() netflix_print(3.4, 0, temp1) self.assertEqual(temp1.getvalue(), "3.4\n")
def test_netflix_print_3(self): w = StringIO.StringIO() ans = {1: [1,2,3], 2: [4,5], 6: [2] } netflix_print(w, ans, [6,2,1]) self.assert_(w.getvalue() == "6:\n2\n2:\n4\n5\n1:\n1\n2\n3\n")
def test_print_3(self): """ tests printing RMSE """ rating = "RMSE: 0.93" writer = StringIO() netflix_print(writer, rating) self.assertEqual(writer.getvalue(), "RMSE: 0.93\n")
def test_print3(self): """ Test printing RMSE format """ writer = StringIO() netflix_print(writer, 0.9722263548654, "RMSE") self.assertEqual(writer.getvalue(), "RMSE: 0.97\n")
def test_print2(self): """ Test printing rating guess format """ writer = StringIO() netflix_print(writer, 3.721318657451, "Rating Guess") self.assertEqual(writer.getvalue(), "3.7\n")
def test_print_1(self): w = StringIO() i = {1: [1, 2, 3]} p = {(3, 1): 3, (1, 1): 2, (2, 1): 1} r = 0 netflix_print(i, p, r, w) self.assertEqual(w.getvalue(), "1:\n2.0\n1.0\n3.0\nRMSE: 0.00\n")
def test_print_1(self): """ Tests print to see if prints a movie line correctly """ writer = StringIO() netflix_print(writer, True, 2153330) self.assertEqual(writer.getvalue(), "2153330:\n")
def test_print_3(self): """ Tests print to see if prints a prediction line correctly """ writer = StringIO() netflix_print(writer, False, 3.59648694869) self.assertEqual(writer.getvalue(), "3.6\n")
def test_print1(self): """ Test printing movie format """ writer = StringIO() netflix_print(writer, "6478:\n", "Movie") self.assertEqual(writer.getvalue(), "6478:\n")
def test_netflix_print_0(self): w = StringIO() movie_id = 1 customer_ids = [1, 2, 3] ratings = [3, 3, 3] netflix_print(movie_id, customer_ids, ratings, w) self.assertEqual(w.getvalue(), "1:\n3.0\n3.0\n3.0\n")
def test_print_2(self): w = StringIO() i = {1: [1, 2], 2: [1, 2]} p = {(1, 2): 4.1, (1, 1): 3, (2, 1): 1.89, (2, 2): 3} r = 0.63 netflix_print(i, p, r, w) self.assertEqual(w.getvalue(), "1:\n3.0\n1.9\n2:\n4.1\n3.0\nRMSE: 0.63\n")
def test_netflix_print_5(self): """ Test 5 netflix_print to see if the rmse is formatted correctly """ write = StringIO() netflix_print(write, .9849846, 'rmse') self.assertEqual(write.getvalue(), 'RMSE: 0.98\n')
def test_print_3(self): w = StringIO() i = {1: [1], 2: [1], 3: [1]} p = {(1, 2): 2, (1, 3): 2.25, (1, 1): 0.76} r = 0.93 netflix_print(i, p, r, w) self.assertEqual(w.getvalue(), "1:\n0.8\n2:\n2.0\n3:\n2.2\nRMSE: 0.93\n")
def test_netflix_print_2(self): """ Test 2 netflix_print to see if movies are formatted correctly """ write = StringIO() netflix_print(write, '199999:', 'movie') self.assertEqual(write.getvalue(), '199999:\n')
def test_print_4(self): ''' Testing the netflix_print() function by passing a StringIO object to write to instead of standard out. ''' write_io = StringIO() netflix_print(write_io, 21, [1, 5]) self.assertEqual(write_io.getvalue(), "21:\n1\n5\n")
def test_netflix_print_3(self): """ Test 3 netflix_print to see if ratings are formatted correctly """ write = StringIO() netflix_print(write, 3.5456465, 'rating') self.assertEqual(write.getvalue(), '3.5\n')
def test_netflix_print_2(self): """ netflix_print_2 """ input_dic = {1: {1000, 1001, 1002, 1003}} store_rating = {1: {3.4, 3.5, 3.6, 3.7}} writer = StringIO() netflix_print(writer, input_dic, store_rating) self.assertEqual(writer.getvalue(), "1:\n1000 3.6\n1001 3.5\n1002 3.7\n1003 3.4\n")
def test_netflix_print_1(self): """ netflix_print_1 """ input_dic = {1: {2001}, 2: {2002}, 3: {2003}} store_rating = {1: {3.5}, 2: {3.6}, 3: {3.7}} writer = StringIO() netflix_print(writer, input_dic, store_rating) self.assertEqual(writer.getvalue(), "1:\n2001 3.5\n2:\n2002 3.6\n3:\n2003 3.7\n")
def test_netflix_print_1(self): w = StringIO() movie_id = 7 customer_ids = [3, 2, 1] ratings = [5.1, 4.2, 3.3] netflix_print(movie_id, customer_ids, ratings, w) self.assertEqual(w.getvalue(), "7:\n5.1\n4.2\n3.3\n")
def test_netflix_print_4(self): """ Test 4 netflix_print to see if ratings are formatted correctly """ write = StringIO() netflix_print(write, 1, 'rating') self.assertEqual(write.getvalue(), '1.0\n')
def test_netflix_print_6(self): """ Test 6 netflix_print to see if the rmse is formatted correctly """ write = StringIO() # NOTE: We round instead of truncate. Is this permissable? netflix_print(write, .555555, 'rmse') self.assertEqual(write.getvalue(), 'RMSE: 0.56\n')
def test_print_0(self): """ Test the printing functionallity in Netflix.py """ writer = StringIO() netflix_print(writer, "1", { 30878: 1, 2647871: 2, 123744: 3 }, [30878, 2647871, 123744]) self.assertEqual(writer.getvalue(), "1:\n1\n2\n3\n")
def test_print_1(self): """ Test the printing functionallity in Netflix.py """ writer = StringIO() netflix_print(writer, "150", { 123: 5, 456: 2, 938: 1, 9999: 0 }, [9999, 938, 456, 123]) self.assertEqual(writer.getvalue(), "150:\n0\n1\n2\n5\n")
def test_print_4 (self) : w = StringIO() netflix_print(w, 1904905) self.assertEqual(w.getvalue(), "1904905 \n")
def test_netflix_print_1 (self) : w = StringIO.StringIO() netflix_print(w, "1") self.assert_(w.getvalue() == "1\n")
def test_print_3 (self) : w = StringIO() netflix_print(w, 3.7) self.assertEqual(w.getvalue(), "3.7\n")
def test_print_3 (self): #no predictions w = StringIO() print_data = ["2043:"] rms_error = .8501 netflix_print(w, print_data, rms_error) self.assertEqual(w.getvalue(), "2043:\nRMSE:0.8501\n")
def test_print_1 (self) : w = StringIO() print_data = ["2043:", 1.5, "4014:", 2.5] #two sets rms_error = .9999 netflix_print(w, print_data, rms_error) self.assertEqual(w.getvalue(), "2043:\n1.5\n4014:\n2.5\nRMSE:0.9999\n")
def test_print_1 (self) : w = StringIO() predictions_dict = OrderedDict([(2043, [3.4, 4.1, 1.9])]) netflix_print(w, predictions_dict) self.assertEqual(w.getvalue(), "2043:\n3.4\n4.1\n1.9\n")
def test_print_2 (self) : output_stream = StringIO() output_dictionary = {1 : (1, 2, 3), 2 : (1, 2, 3)} rmse_value = 1 netflix_print(output_stream, output_dictionary, rmse_value) self.assertEqual("1:\n1.0\n2.0\n3.0\n2:\n1.0\n2.0\n3.0\nRMSE: 1.00\n", output_stream.getvalue())
def test_print_3(self): w = StringIO() netflix_print(w, 90, []) self.assertEqual(w.getvalue(), "90:\n")
def test_print_3 (self) : output_stream = StringIO() output_dictionary = {} rmse_value = 1 netflix_print(output_stream, output_dictionary, rmse_value) self.assertEqual("RMSE: 1.00\n", output_stream.getvalue())
def test_print_1 (self) : w = StringIO() netflix_print(w, 1.9) self.assertEqual(w.getvalue(), "1.9 \n")
def test_print_5 (self) : w = StringIO() netflix_print(w, 9293277) self.assertEqual(w.getvalue(), "9293277\n")
def test_print_1 (self) : w = StringIO() netflix_print(w, 3423) self.assertEqual(w.getvalue(), "3423\n")
def test_print_2 (self) : w = StringIO() netflix_print(w, 92838) self.assertEqual(w.getvalue(), "92838\n")
def test_netflix_print_2 (self) : w = StringIO.StringIO() netflix_print(w, 4) self.assert_(w.getvalue() == "4\n")
def test_print_2(self): """ Test the printing functionallity in Netflix.py """ writer = StringIO() netflix_print(writer, "5", {53: 3, 827: 2, 128: 4}, [827, 128, 53]) self.assertEqual(writer.getvalue(), "5:\n2\n4\n3\n")
def test_print_1(self): w = StringIO() netflix_print(w, LineType.movie, 111) self.assertEqual(w.getvalue(), "111:\n")
def test_print_3 (self) : w = StringIO() predictions_dict = OrderedDict([(2043, [3.4, 4.1, 1.9]), (10851, [4.3, 1.4, 2.8]), (3367, [3.2, 1.1, 4.5])]) netflix_print(w, predictions_dict) self.assertEqual(w.getvalue(), "2043:\n3.4\n4.1\n1.9\n10851:\n4.3\n1.4\n2.8\n3367:\n3.2\n1.1\n4.5\n")
def test_print_2(self): w = StringIO() netflix_print(w, LineType.user, 4.31) self.assertEqual(w.getvalue(), "4.3\n")
def test_print_2 (self): #one set w = StringIO() print_data = ["2043:", 1.5, 4.5] rms_error = .8501 netflix_print(w, print_data, rms_error) self.assertEqual(w.getvalue(), "2043:\n1.5\n4.5\nRMSE:0.8501\n")
def test_print3 (self) : w = StringIO() netflix_print(w, ["1:","17770:",2]) self.assertEqual(w.getvalue(), "1:\n17770:\n2\n")
def test_print_4 (self) : #no movie number w = StringIO() print_data = [1.5, 4.5] rms_error = .8501 netflix_print(w, print_data, rms_error) self.assertEqual(w.getvalue(), "1.5\n4.5\nRMSE:0.8501\n")
def test_print2 (self) : w = StringIO() netflix_print(w, ["1:",3,4]) self.assertEqual(w.getvalue(), "1:\n3\n4\n")
def test_netflix_print_2(self): w = StringIO.StringIO() ans = {} netflix_print(w, ans, []) self.assert_(w.getvalue() == "")
def test_print_2 (self) : w = StringIO() netflix_print(w, 2.79) self.assertEqual(w.getvalue(), "2.79 \n")
def test_print_1 (self) : w = StringIO() netflix_print(w, 4.279008432960954) self.assertEqual(w.getvalue(), "4.279008432960954\n")
def test_print_3 (self) : w = StringIO() netflix_print(w, 11134234) self.assertEqual(w.getvalue(), "11134234\n")
def test_print_2 (self) : w = StringIO() netflix_print(w, 3.4756) self.assertEqual(w.getvalue(), "3.4756\n")
def test_netflix_print_3 (self) : w = StringIO.StringIO() netflix_print(w, "123412:") self.assert_(w.getvalue() == "123412:\n")