def test_get_boundary_counts(kwik_e_mart_nlp, expected, predicted, expected_counts): predicted_counts = taggers.get_boundary_counts( expected, predicted, taggers.BoundaryCounts()).to_dict() for key in predicted_counts.keys(): if predicted_counts[key] != expected_counts.get(key, 0): print("predicted counts: {}".format(predicted_counts)) assert False
def test_get_boundary_counts_sequential(kwik_e_mart_nlp, expected, predicted, expected_counts): boundary_counts = taggers.BoundaryCounts() for expected_sequence, predicted_sequence in zip(expected, predicted): boundary_counts = taggers.get_boundary_counts(expected_sequence, predicted_sequence, boundary_counts) predicted_counts = boundary_counts.to_dict() for key in predicted_counts.keys(): if predicted_counts[key] != expected_counts.get(key, 0): assert False