def setUp(self):
db = bsddb.btopen(None, cachesize=268435456)
matrix = DictMatrix(10, 10, db)
m1, m2 = ([0, 1, 2, 3], [2, 3, 1, 4])
for i in range(1, 3):
for j in range(1, 5):
if i == 1:
matrix[i, j] = m1[j - 1]
else:
matrix[i, j] = m2[j - 1]
self.similar = UserSimilar(matrix)
class UserSimilarTestCase(unittest.TestCase):
def setUp(self):
db = bsddb.btopen(None, cachesize=268435456)
matrix = DictMatrix(10, 10, db)
m1, m2 = ([0, 1, 2, 3], [2, 3, 1, 4])
for i in range(1, 3):
for j in range(1, 5):
if i == 1:
matrix[i, j] = m1[j - 1]
else:
matrix[i, j] = m2[j - 1]
self.similar = UserSimilar(matrix)
def tearDown(self):
self.similar = None
def testCompute(self):
self.assertEqual(True, abs(self.similar.compute(1, 2) - 0.166) < 0.01)
def testTopNNoWeight(self):
self.assertEqual([2], self.similar.testTopNNoWeight(1, 1))
