Added test for imports.

NowanIlfideme · Sep 13, 2021 · 97daf9c · 97daf9c
1 parent 80cf781
commit 97daf9c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/test_import.py b/test/test_import.py
@@ -4,3 +4,12 @@
 def test_import_basic():
     """This tests whether the `tsdata` package is importable."""
     import tsdata  # noqa
+
+
+def test_import_raw():
+    """This tests whether the `tsdata` package loads a dataset."""
+    from tsdata.raw import available_data, load_data
+
+    assert len(available_data()) > 0
+    assert "LakeHuron" in available_data()
+    load_data("LakeHuron")