def test_apriori(self): # Assert frequent sets frequency. v = vector.apriori(([1, 2, 4], [1, 2, 5], [1, 3, 6], [1, 3, 7]), support=0.5) self.assertTrue(len(v), 3) self.assertEqual(v[frozenset((1,))], 1.0) self.assertEqual(v[frozenset((1, 2))], 0.5) self.assertEqual(v[frozenset((2,))], 0.5) self.assertEqual(v[frozenset((3,))], 0.5)
def test_apriori(self): # Assert frequent sets frequency. v = vector.apriori(([1, 2, 4], [1, 2, 5], [1, 3, 6], [1, 3, 7]), support=0.5) self.assertTrue(len(v), 3) self.assertEqual(v[frozenset((1, ))], 1.0) self.assertEqual(v[frozenset((1, 2))], 0.5) self.assertEqual(v[frozenset((2, ))], 0.5) self.assertEqual(v[frozenset((3, ))], 0.5)