def test_combine_measures():
CALC = MeasureCalculator()
assert CALC.combine_measures(first_set, second_set) == {
"BernoulliNB": {"cross score": 3, "precision": 1, "recall": 3, "f1": 1, "cross variance": 1},
"MultinomialNB": {"cross score": 2, "precision": 2, "recall": 2, "f1": 2, "cross variance": 2},
}
def test_add_measures():
CALC = MeasureCalculator()
CALC.add_measures(first_set, second_set)
assert CALC.measures == {
"BernoulliNB": {"cross score": [3], "precision": [1], "recall": [3], "f1": [1], "cross variance": [1]},
"MultinomialNB": {"cross score": [2], "precision": [2], "recall": [2], "f1": [2], "cross variance": [2]},
}