From abe9734608c04de9b3a464ac30ffb5ba48371db0 Mon Sep 17 00:00:00 2001 From: sgbaird Date: Mon, 14 Nov 2022 13:42:33 -0700 Subject: [PATCH] only test fetch_data if running on Python >= 3.8 --- tests/test_mp_time_split.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/test_mp_time_split.py b/tests/test_mp_time_split.py index 670c4cc..a5d1094 100644 --- a/tests/test_mp_time_split.py +++ b/tests/test_mp_time_split.py @@ -1,5 +1,7 @@ +import sys from os import path +import pytest from matminer.utils.io import load_dataframe_from_json from mp_time_split.core import MPTimeSplit, get_data_home @@ -22,6 +24,7 @@ # ) +@pytest.mark.skipif(sys.version_info < (3, 8), reason="requires Python 3.8+") def test_data_snapshot_one_by_one(): dummy_expt_df_check = load_dataframe_from_json(dummy_data_path) mpt = MPTimeSplit(num_sites=num_sites, elements=elements) @@ -33,6 +36,7 @@ def test_data_snapshot_one_by_one(): ) +@pytest.mark.skipif(sys.version_info < (3, 8), reason="requires Python 3.8+") def test_get_train_and_val_data(): mpt = MPTimeSplit(num_sites=num_sites, elements=elements) mpt.fetch_data(one_by_one=True) @@ -51,6 +55,7 @@ def test_get_train_and_val_data(): return train_inputs, val_inputs, train_outputs, val_outputs +@pytest.mark.skipif(sys.version_info < (3, 8), reason="requires Python 3.8+") def test_get_test_data(): mpt = MPTimeSplit(num_sites=num_sites, elements=elements) mpt.fetch_data(one_by_one=True)