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