Initial build out

tests/test_ipeds_enrollment.py

@@ -0,0 +1,25 @@
+import shutil
+
+import admin_analytics.config as config
+from admin_analytics.config import UD_UNITID
+from admin_analytics.ipeds.enrollment import load_enrollment
+
+
+def test_load_enrollment_filters_to_ud(db_conn, fixtures_dir, tmp_path):
+    ipeds_dir = tmp_path / "ipeds" / "enrollment" / "2023"
+    ipeds_dir.mkdir(parents=True)
+    shutil.copy(fixtures_dir / "ef2023.csv", ipeds_dir / "ef2023.csv")
+
+    original = config.IPEDS_DATA_DIR
+    config.IPEDS_DATA_DIR = tmp_path / "ipeds"
+    try:
+        count = load_enrollment(db_conn, range(2023, 2024), unitid_filter=UD_UNITID)
+    finally:
+        config.IPEDS_DATA_DIR = original
+
+    assert count == 1
+    row = db_conn.execute(
+        "SELECT total_enrollment FROM raw_ipeds_enrollment WHERE unitid = ?",
+        [UD_UNITID],
+    ).fetchone()
+    assert row[0] == 24120