def test_empty_init(self):
label_encoder = data.LabelEncoder()
assert label_encoder.index_to_class == {}
assert len(label_encoder.classes) == 0
def test_dict_init(self):
class_to_index = {"apple": 0, "banana": 1}
label_encoder = data.LabelEncoder(class_to_index=class_to_index)
assert label_encoder.index_to_class == {0: "apple", 1: "banana"}
assert len(label_encoder.classes) == 2
def setup_method(self):
"""Called before every method."""
self.label_encoder = data.LabelEncoder()