def test_process_labels_warning(self):
labels = ['a', 'b']
# this function just looks at lists, not the actual contents type
dummy_taxonomies = [1, 2, 3, 4, 5]
with self.assertWarnsRegex(
UserWarning, "taxonomies and labels are different lengths"):
new_labels = evaluate._process_labels(labels, dummy_taxonomies)
self.assertEqual(new_labels, ['a', 'b', 3, 4, 5])
def test_process_labels_empty(self):
dummy_taxonomies = [1, 2, 3]
new_labels = evaluate._process_labels(None, dummy_taxonomies)
self.assertEqual(new_labels, [1, 2, 3])
def test_process_labels_too_many_labels(self):
labels = ['a', 'b', 'c', 'd', 'e']
dummy_taxonomies = [1, 2, 3]
new_labels = evaluate._process_labels(labels, dummy_taxonomies)
self.assertEqual(new_labels, ['a', 'b', 'c'])