fixing unit tests

Author: manu-sj

python/tests/utils/schema/test_columnar_schema.py

@@ -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
@@ -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