def test_movie_avg3(self): trainingSetDir = "test/5_movies_5_cust/" w = StringIO.StringIO() netflix_movie_avg(trainingSetDir, w) self.assert_(w.getvalue() == '1=1.0\n2=2.0\n3=3.0\n4=4.0\n5=5.0\n')
def test_movie_avg(self): trainingSetDir = "test/1_movie_1_cust/" w = StringIO.StringIO() netflix_movie_avg(trainingSetDir, w) self.assert_(w.getvalue() == '1=1.0\n')