diff --git a/tests/plugin/text/test_dob_plugin/test_dob_plugin.py b/tests/plugin/text/test_dob_plugin/test_dob_plugin.py index 1dbd7264..03fa18f0 100644 --- 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