Skip to content

Commit

Permalink
fix index issue in Python 3.8
Browse files Browse the repository at this point in the history
  • Loading branch information
danlu1 committed Nov 14, 2024
1 parent c2696e5 commit 9d31715
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions tests/test_process_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,12 @@
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(
Expand Down Expand Up @@ -655,6 +659,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"]
)
result.reset_index(drop=True, inplace=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"]
Expand Down

0 comments on commit 9d31715

Please sign in to comment.