def test_get_bug_labels():
model = DefectModel()
classes, _ = model.get_labels()
# labels from bug_nobug.csv
assert classes[1087488]
assert not classes[1101825]
# labels from regression_bug_nobug.csv
assert not classes[447581] # nobug
assert classes[518272] # regression
assert classes[528988] # bug_unknown_regression
assert classes[1037762] # bug_no_regression
# labels from defectenhancementtask.csv
assert not classes[1488307] # task
assert classes[1488310] # defect
assert not classes[1531080] # enhancement
def __init__(self, lemmatization=False, historical=False):
DefectModel.__init__(self, lemmatization, historical)
self.calculate_importance = False
def __init__(self, lemmatization=False):
DefectModel.__init__(self, lemmatization)
def __init__(self, lemmatization=False, historical=False):
DefectModel.__init__(self, lemmatization, historical)
def test_get_defect_labels():
model = DefectModel()
classes, _ = model.get_labels()
assert classes[1042414] == 1
assert classes[1049816] != 1