Update test

astrolabsoftware · Oct 24, 2024 · da16512 · da16512
1 parent 4cb73fe
commit da16512
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/api_ssoft_test.py b/tests/api_ssoft_test.py
@@ -162,6 +162,18 @@ def compare_schema() -> None:
     assert keys1 == keys2, [keys1, keys2]
 
 
+def check_sshg1g2() -> None:
+    """
+    Examples
+    --------
+    >>> check_sshg1g2()
+    """
+    pdf = ssoftsearch(flavor="SSHG1G2")
+
+    assert "period" in pdf.columns
+    assert "a_b" in pdf.columns
+
+
 if __name__ == "__main__":
     """ Execute the test suite """
     import sys