From 176eed77bfbd30ddffce58744c0e5028d6c53835 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Fri, 2 Feb 2024 11:20:39 +0100 Subject: [PATCH] chg: remove IntEnum --- pymisp/abstract.py | 4 ++-- tests/testlive_comprehensive.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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)