diff options
| -rw-r--r-- | tests/test_tableflow.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/tests/test_tableflow.py b/tests/test_tableflow.py index 41f3620..7d655b1 100644 --- a/tests/test_tableflow.py +++ b/tests/test_tableflow.py @@ -26,7 +26,7 @@ def test_right_comment_in_table_file_to_load(): outputFileName=None, comment="#", ) - assert type(tIn) == pd.core.frame.DataFrame + assert isinstance(tIn, pd.core.frame.DataFrame) def test_table_equality_with_no_output_file_name(): @@ -35,8 +35,8 @@ def test_table_equality_with_no_output_file_name(): outputFileName=None, comment="#", ) - assert type(tIn) == pd.core.frame.DataFrame - assert type(tOut) == pd.core.frame.DataFrame + assert isinstance(tIn, pd.core.frame.DataFrame) + assert isinstance(tOut, pd.core.frame.DataFrame) assert tIn.equals(tOut) col0 = tIn.keys()[0] vBefore = tIn.at[0, col0] @@ -51,8 +51,8 @@ def test_table_load_with_in_out_file_names(): outputFileName="tests/tableflow_test_data/tableOut1nonProcessed.csv", comment="#", ) - assert type(tIn) == pd.core.frame.DataFrame - assert type(tOut) == pd.core.frame.DataFrame + assert isinstance(tIn, pd.core.frame.DataFrame) + assert isinstance(tOut, pd.core.frame.DataFrame) assert tIn.equals(tOut) # different filenames, different content @@ -61,8 +61,8 @@ def test_table_load_with_in_out_file_names(): outputFileName="tests/tableflow_test_data/tableOut1pariallyProcessed.csv", comment="#", ) - assert type(tIn) == pd.core.frame.DataFrame - assert type(tOut) == pd.core.frame.DataFrame + assert isinstance(tIn, pd.core.frame.DataFrame) + assert isinstance(tOut, pd.core.frame.DataFrame) assert not tIn.equals(tOut) assert "out1" in tOut.columns assert "out1" not in tIn.columns |