Add some tests

Author: nathanjmcdougall

pins/tests/test_adaptors.py

@@ -27,6 +27,12 @@ def test_non_df(self):
         assert not isinstance(adaptor, PandasAdaptor)
         assert not isinstance(adaptor, DFAdaptor)
 
+    def test_already_adaptor(self):
+        df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]})
+        adaptor = create_adaptor(df)
+        assert isinstance(adaptor, PandasAdaptor)
+        assert create_adaptor(adaptor) is adaptor
+
 
 class TestAdaptor:
     def test_write_json(self, tmp_path: Path):

pins/tests/test_drivers.py

@@ -6,6 +6,7 @@
 import pandas as pd
 import pytest
 
+from pins._adaptors import create_adaptor
 from pins.config import PINS_ENV_INSECURE_READ
 from pins.drivers import default_title, load_data, load_path, save_data
 from pins.errors import PinsInsecureReadError
@@ -163,6 +164,23 @@ def test_driver_apply_suffix_false(tmp_path: Path):
     assert Path(res_fname).name == "some_df"
 
 
+class TestSaveData:
+    def test_accepts_pandas_df(self, tmp_path: Path):
+        import pandas as pd
+
+        df = pd.DataFrame({"x": [1, 2, 3]})
+        result = save_data(df, tmp_path / "some_df", "csv")
+        assert Path(result) == tmp_path / "some_df.csv"
+
+    def test_accepts_adaptor(self, tmp_path: Path):
+        import pandas as pd
+
+        df = pd.DataFrame({"x": [1, 2, 3]})
+        adaptor = create_adaptor(df)
+        result = save_data(adaptor, tmp_path / "some_df", "csv")
+        assert Path(result) == tmp_path / "some_df.csv"
+
+
 class TestLoadFile:
     def test_str_file(self):
         class _MockMetaStrFile: