Switch regex strategy to sampling strategy to improve performance (#2998

) * Switch regex strategy to sampling strategy to improve performance * Increase deadline
catalyst-cooperative · Nov 2, 2023 · c11b1a4 · c11b1a4
1 parent e7aedda
commit c11b1a4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/unit/io_managers_test.py b/test/unit/io_managers_test.py
@@ -354,7 +354,7 @@ def test_ferc_xbrl_sqlite_io_manager_dedupes(mocker, tmp_path):
 example_schema = pandera.DataFrameSchema(
     {
         "entity_id": pandera.Column(
-            str, pandera.Check.str_matches(r"[0-9a-zA-Z]+"), nullable=False
+            str, pandera.Check.isin("C0123456789"), nullable=False
         ),
         "date": pandera.Column("datetime64[ns]", nullable=False),
         "utility_type": pandera.Column(
@@ -365,11 +365,12 @@ def test_ferc_xbrl_sqlite_io_manager_dedupes(mocker, tmp_path):
         "publication_time": pandera.Column("datetime64[ns]", nullable=False),
         "int_factoid": pandera.Column(int),
         "float_factoid": pandera.Column(float),
-        "str_factoid": pandera.Column("str"),
+        "str_factoid": pandera.Column(str),
     }
 )
 
 
+@hypothesis.settings(print_blob=True, deadline=400)
 @hypothesis.given(example_schema.strategy(size=3))
 def test_filter_for_freshest_data(df):
     # XBRL context is the identifying metadata for reported values