diff --git a/pymisp/abstract.py b/pymisp/abstract.py index 068e1a5..41b335a 100644 --- a/pymisp/abstract.py +++ b/pymisp/abstract.py @@ -55,14 +55,14 @@ class Distribution(Enum): inherit = 5 -class ThreatLevel(IntEnum): +class ThreatLevel(Enum): high = 1 medium = 2 low = 3 undefined = 4 -class Analysis(IntEnum): +class Analysis(Enum): initial = 0 ongoing = 1 completed = 2 diff --git a/tests/testlive_comprehensive.py b/tests/testlive_comprehensive.py index 67f6d2b..036cf3a 100644 --- a/tests/testlive_comprehensive.py +++ b/tests/testlive_comprehensive.py @@ -1705,7 +1705,7 @@ class TestComprehensive(unittest.TestCase): first_send = self.create_simple_event() try: first = self.user_misp_connector.add_event(first_send) - self.assertTrue(isinstance(first, MISPEvent), '\n'.join([first_send, first])) + self.assertTrue(isinstance(first, MISPEvent), first) file_obj, bin_obj, sections = make_binary_objects('tests/viper-test-files/test_files/whoami.exe', standalone=False) first.add_object(file_obj) first.add_object(bin_obj)