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]},
    }