def test_counts(self): cols = [1,2] row = [0,1,0] data = [[0,1,1], [0,1,0], [0,0,0], [0,1,1]] classes = ['a', 'b', 'a', 'b'] poss = ['a', 'b'] counts = BNClassifier.counts(cols, row, data, classes, poss) self.assertEquals(1, counts['a']) self.assertEquals(2, counts['b'])