added another test

axiom-data-science · Apr 28, 2023 · 0b991b4 · 0b991b4
1 parent c5dec82
commit 0b991b4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_accessor.py b/tests/test_accessor.py
@@ -139,3 +139,9 @@ def test_index():
     df = pd.DataFrame(index=["m_time"])
     df.index.rename("m_time", inplace=True)
     assert df.cf["T"].name == "m_time"
+
+
+def test_cols():
+    df = pd.DataFrame(columns=["m_time", "lon", "lat", "temp"])
+    assert df.cf.axes_cols == ["m_time"]
+    assert sorted(df.cf.coordinates_cols) == ["lat", "lon", "m_time"]