fix integration tests

Author: francesco086

test/data/load/parquet_as_pandas.py

@@ -4,4 +4,6 @@
 
 
 def load() -> pd.DataFrame:
-    return pd.read_parquet(Path(__file__).parent / "sample.parquet")
+    df = pd.read_parquet(Path(__file__).parent / "sample.parquet")
+    df["datetime1"] = df["datetime1"].astype("datetime64[ns]")
+    return df

test/data/schema/wrong_syntax.py

@@ -2,6 +2,6 @@
 from pandera.typing import Series
 
 
-class Schema(pa.SchemaModel):
+class Schema(pa.DataFrameModel):
     A: Series[int] = pa.Field(nullable=False)
     B: Series[float] = pa.Field(nullable=False)

test/integration_test/pack_test.py

@@ -4,6 +4,13 @@
 from importlib.metadata import version
 from pathlib import Path
 from tempfile import TemporaryDirectory
+from typing import Generator, Optional
+
+import pandas as pd
+import pandera as pa
+import pytest
+
+from dac import PackConfig, PyProjectConfig
 from test.cli_utilities import invoke_dac_pack_from_config
 from test.data import (
     generate_random_project_name,
@@ -18,12 +25,6 @@
     get_test_project_version,
 )
 from test.data.load import self_contained_as_pandas
-from typing import Generator, Optional
-
-import pandas as pd
-import pandera as pa
-import pytest
-from dac import PackConfig, PyProjectConfig
 
 
 @pytest.mark.slow
@@ -54,7 +55,7 @@ def test_if_installed_then_can_access_schema():
         files_in_wheel_dir = list(Path(config.wheel_dir).iterdir())
         with _pip_installed_wheel(path=files_in_wheel_dir[0]):
             pkg = import_module(config.pyproject.project_name)
-            assert issubclass(pkg.Schema, pa.SchemaModel)
+            assert issubclass(pkg.Schema, pa.DataFrameModel)
 
 
 @pytest.mark.slow