Add test

misi9170 · Jul 26, 2024 · 49a0e58 · 49a0e58
1 parent 2714c2b
commit 49a0e58
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/flasc_dataframe_test.py b/tests/flasc_dataframe_test.py
@@ -0,0 +1,11 @@
+import pandas as pd
+
+from flasc.flasc_dataframe import FlascDataFrame
+
+
+def test_type():
+    df = FlascDataFrame({"a": [1, 2, 3], "b": [4, 5, 6], "c": [7, 8, 9]}, name_map={"a": "AA"})
+    assert isinstance(df, FlascDataFrame)
+
+    # Assert df is a pandas DataFrame
+    assert isinstance(df, pd.DataFrame)