From ff7fe217b25a1e86e0d52de775b9868108348b9d Mon Sep 17 00:00:00 2001
From: emilyhunt <emily.hunt.physics@gmail.com>
Date: Thu, 25 Sep 2025 16:51:39 +0200
Subject: [PATCH] Forgot to reset data_dir

---
 tests/test_data.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/test_data.py b/tests/test_data.py
index 4411299..fa2cdad 100644
--- a/tests/test_data.py
+++ b/tests/test_data.py
@@ -135,5 +135,9 @@ def test_set_data_directory():
         assert _CONFIG["first_run"]
         assert _CONFIG["data_already_downloaded"] is False
 
+        # Now reset it
+        set_data_directory(_DEFAULT_DIRECTORY)
+        assert _CONFIG["data_dir"] == _DEFAULT_DIRECTORY
+
     finally:
         shutil.rmtree(location)
-- 
2.51.2