/
TestParsers.py
88 lines (70 loc) · 2.68 KB
/
TestParsers.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#!/usr/bin/env python
# ---------------------------
# CS 373 Project 3
# Anh-Khoi Dang - add562
# ---------------------------
# -------
# imports
# -------
import StringIO
import unittest
from Netflix import netflix_parse_precomputed, netflix_decade_calc
from Parsers import netflix_decade_avg, netflix_movie_avg, netflix_cust_avg
# -----------
# TestNetflix
# -----------
class TestParsers (unittest.TestCase) :
# --------
# Parsers
# --------
def test_decade_avg(self):
trainingSetDir = "test/1_movie_1_cust/"
w = StringIO.StringIO()
netflix_decade_avg(trainingSetDir, w)
self.assert_(w.getvalue() == '1:\n2000s=1.0\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')
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_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')
def test_movie_avg2(self):
trainingSetDir = "test/5_movies_1_cust/"
w = StringIO.StringIO()
netflix_movie_avg(trainingSetDir, w)
self.assert_(w.getvalue() == '1=1.0\n2=1.0\n3=1.0\n4=1.0\n5=1.0\n')
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_cust_avg(self):
trainingSetDir = "test/1_movie_1_cust/"
w = StringIO.StringIO()
netflix_cust_avg(trainingSetDir, w)
self.assert_(w.getvalue() == '1=1.0\n')
def test_cust_avg2(self):
trainingSetDir = "test/5_movies_1_cust/"
w = StringIO.StringIO()
netflix_cust_avg(trainingSetDir, w)
self.assert_(w.getvalue() == '1=1.0\n')
def test_cust_avg3(self):
trainingSetDir = "test/5_movies_5_cust/"
w = StringIO.StringIO()
netflix_cust_avg(trainingSetDir, w)
self.assert_(w.getvalue() == '100=3.0\n200=3.0\n300=3.0\n400=3.0\n500=3.0\n')
# ----
# main
# ----
print "TestParsers.py"
unittest.main()
print "Done."