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()