wrote test function to cover the 'transform' function in DOBPlugin to…

… attain full coverage
skit-ai · Jan 9, 2024 · ad5b8db · ad5b8db
1 parent 068e8ec
commit ad5b8db
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/plugin/text/test_dob_plugin/test_dob_plugin.py b/tests/plugin/text/test_dob_plugin/test_dob_plugin.py
@@ -51,3 +51,15 @@ async def test_get_transcripts_from_utterances(test_case) -> None:
         pass
 
 
+
+@pytest.mark.asyncio
+async def test_invalid_data() -> None:
+    print("test 3")
+    train_df = pd.DataFrame(
+        [
+            {"data": json.dumps([[{"transcript": "yes"}]])},
+            {"data": json.dumps({})},
+        ]
+    )
+    train_df_ = await get_dob.transform(train_df)
+    assert len(train_df) - len(train_df_) == 1