def test_decade_avg3(self): trainingSetDir = "test/5_movies_5_cust/" w = StringIO.StringIO() netflix_decade_avg(trainingSetDir, w) self.assert_(w.getvalue() == '100:\n1990s=3.5\n2000s=2.66666666667\n200:\n1990s=3.5\n2000s=2.66666666667\n300:\n1990s=3.5\n2000s=2.66666666667\n400:\n1990s=3.5\n2000s=2.66666666667\n500:\n1990s=3.5\n2000s=2.66666666667\n')
def test_decade_avg2(self): trainingSetDir = "test/5_movies_1_cust/" w = StringIO.StringIO() netflix_decade_avg(trainingSetDir, w) self.assert_(w.getvalue() == '1:\n1990s=1.0\n2000s=1.0\n')