moved tokenization tests to all be in a single function rather than two

mmcdermott · Sep 1, 2024 · df86efb · df86efb
1 parent 1842365
commit df86efb
Showing 1 changed file with 0 additions and 11 deletions.
diff --git a/tests/MEDS_Transforms/test_tokenization.py b/tests/MEDS_Transforms/test_tokenization.py
@@ -302,8 +302,6 @@ def test_tokenization():
         should_error=True,
     )
 
-
-def test_tokenization_missing_static():
     single_stage_transform_tester(
         transform_script=TOKENIZATION_SCRIPT,
         stage_name="tokenization",
@@ -312,12 +310,3 @@ def test_tokenization_missing_static():
         want_data={**WANT_SCHEMAS_MISSING_STATIC, **WANT_EVENT_SEQS},
         df_check_kwargs={"check_column_order": False},
     )
-
-    single_stage_transform_tester(
-        transform_script=TOKENIZATION_SCRIPT,
-        stage_name="tokenization",
-        transform_stage_kwargs={"train_only": True},
-        input_shards=NORMALIZED_SHARDS_MISSING_STATIC,
-        want_data={**WANT_SCHEMAS_MISSING_STATIC, **WANT_EVENT_SEQS},
-        should_error=True,
-    )