From c2696e592dfdcf7ac384ed562c9e1212fe74d661 Mon Sep 17 00:00:00 2001 From: danlu1 Date: Thu, 14 Nov 2024 00:41:29 +0000 Subject: [PATCH] fix index issue in testing --- tests/test_process_functions.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/tests/test_process_functions.py b/tests/test_process_functions.py index 7879990a..6c722708 100644 --- 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"]