fix utest

Signed-off-by: Massy Bourennani <[email protected]>

Author: MassyB

core/setup.py

@@ -82,6 +82,8 @@
         "typing-extensions>=4.4",
         "pydantic<2",
         # ----
+        # OpenLineage Dependencies
+        "openlineage-python==1.30.1",
     ],
     zip_safe=False,
     classifiers=[

dev-requirements.txt

@@ -39,3 +39,5 @@ types-pytz
 types-requests
 types-setuptools
 mocker
+# OpenLinage dependency
+openlineage-python==1.30.1

tests/unit/test_events.py

@@ -492,6 +492,7 @@ def test_event_codes(self):
     core_types.ArtifactUploadSuccess(),
     core_types.ArtifactUploadError(),
     core_types.ArtifactUploadSkipped(),
+    core_types.OpenLineageException(),
 ]
 
 
@@ -508,7 +509,7 @@ def test_all_serializable(self):
         diff = all_non_abstract_events.difference(set(all_event_values_list))
         assert (
             not diff
-        ), f"{diff}test is missing concrete values in `sample_values`. Please add the values for the aforementioned event classes"
+        ), f"{diff} test is missing concrete values in `sample_values`. Please add the values for the aforementioned event classes"
 
         # make sure everything in the list is a value not a type
         for event in sample_values: