fix index issue in testing

Sage-Bionetworks · Nov 14, 2024 · c2696e5 · c2696e5
1 parent 49366dc
commit c2696e5
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/tests/test_process_functions.py b/tests/test_process_functions.py
@@ -5,12 +5,8 @@
 import pytest
 import synapseclient
 from genie import process_functions
-from pandas.api.types import (
-    is_bool_dtype,
-    is_float_dtype,
-    is_integer_dtype,
-    is_string_dtype,
-)
+from pandas.api.types import (is_bool_dtype, is_float_dtype, is_integer_dtype,
+                              is_string_dtype)
 from pandas.testing import assert_frame_equal
 
 DATABASE_DF = pd.DataFrame(
@@ -659,7 +655,7 @@ def test_that_create_missing_columns_gets_expected_output_with_single_col_df(
     result = process_functions.create_missing_columns(
         dataset=test_cases["test_input"], schema=test_cases["test_schema"]
     )
-    assert_frame_equal(result, test_cases["expected_output"], check_exact=True)
+    assert_frame_equal(result, test_cases["expected_output"], check_dtype=False)
     assert test_cases["expected_dtype"](result.iloc[:, 0])
     assert result.isna().sum().sum() == test_cases["expected_na_count"]