From 79276fba09d60f329227951fbbe878f8caccd915 Mon Sep 17 00:00:00 2001 From: Drew Oldag Date: Thu, 26 Sep 2024 13:17:40 -0700 Subject: [PATCH] Using example test data for test_learn_loop. --- tests/conftest.py | 4 ++++ tests/test_learn_loop.py | 4 ++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index b3f5319e..eb681c86 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -12,6 +12,10 @@ def test_data_path(): return Path(__file__).parent.parent / "data" / "tests" +@pytest.fixture +def test_des_data_path(): + return Path(__file__).parent.parent / "data" / "tests" / "DES_data" + @pytest.fixture(scope="session") def base_temp(tmp_path_factory): diff --git a/tests/test_learn_loop.py b/tests/test_learn_loop.py index a6b20f5f..110a4fde 100644 --- a/tests/test_learn_loop.py +++ b/tests/test_learn_loop.py @@ -5,13 +5,13 @@ from resspect import fit_snpcc from resspect.learn_loop import learn_loop -def test_can_run_learn_loop(test_data_path): +def test_can_run_learn_loop(test_des_data_path): """Test that learn_loop can load data and run.""" with tempfile.TemporaryDirectory() as dir_name: # Create the feature files to use for the learning loop. output_file = os.path.join(dir_name, "output_file.dat") fit_snpcc( - path_to_data_dir=test_data_path, + path_to_data_dir=test_des_data_path, features_file=output_file, )