Update test_generator.py

george0st · Aug 14, 2024 · a8294c8 · a8294c8
1 parent 6fc122c
commit a8294c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_generator.py b/tests/test_generator.py
@@ -156,6 +156,6 @@ def test_csv_structure(self):
         self._check_csv_header(path.join(dir, f"{basic_event.BasicEvent.NAME}.csv"),
                                ["party_id", "event_id", "session_id"])
         self._check_csv_header(path.join(dir, f"{basic_communication.BasicCommunication.NAME}.csv"),
-                               ["party_id", "communication_id", "content"])
+                               ["party_id", "communication_id", "content", "content_sentiment"])
 
     # TODO: Add batch size under limit, it will generate wrong dataset