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')
