Skip to content

Commit

Permalink
fixing unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
manu-sj committed Oct 31, 2024
1 parent 4a5e174 commit b51cdc9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/tests/utils/schema/test_columnar_schema.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ def test_constructor_default(self, mocker):
mock_convert_pandas_series_to_schema.assert_not_called()
mock_convert_spark_to_schema.assert_not_called()
mock_convert_td_to_schema.assert_not_called()
assert mock_find_spec.call_count == 2
assert mock_find_spec.call_count == 1

def test_constructor_list(self, mocker):
# Arrange
Expand Down Expand Up @@ -257,7 +257,7 @@ def test_constructor_hsfs_td(self, mocker):
mock_convert_pandas_series_to_schema.assert_not_called()
mock_convert_spark_to_schema.assert_not_called()
mock_convert_td_to_schema.assert_called_once_with(columnar_obj)
assert mock_find_spec.call_count == 2
assert mock_find_spec.call_count == 1

# convert list to schema

Expand Down

0 comments on commit b51cdc9

Please sign in to comment.