From 300cbba82e48fcccc12f68a93521b7d14415be74 Mon Sep 17 00:00:00 2001 From: Leland McInnes Date: Thu, 10 Feb 2022 09:14:32 -0500 Subject: [PATCH] Skip save load test on windows where things are strange --- umap/tests/test_parametric_umap.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/umap/tests/test_parametric_umap.py b/umap/tests/test_parametric_umap.py index e1e07237..a2b5a554 100644 --- a/umap/tests/test_parametric_umap.py +++ b/umap/tests/test_parametric_umap.py @@ -4,6 +4,7 @@ from sklearn.datasets import make_moons from sklearn.model_selection import train_test_split from numpy.testing import assert_array_almost_equal +import platform try: import tensorflow as tf @@ -15,7 +16,7 @@ from umap.parametric_umap import ParametricUMAP, load_ParametricUMAP tf_only = pytest.mark.skipif(not IMPORT_TF, reason="TensorFlow >= 2.0 is not installed") - +not_windows = pytest.mark.skipif(platform.system() == "Windows", reason="Windows file access issues") @pytest.fixture(scope="session") def moon_dataset(): @@ -124,6 +125,7 @@ def test_validation(moon_dataset): assert embedding.shape == (X_train.shape[0], 2) +@not_windows @tf_only def test_save_load(moon_dataset): """tests saving and loading""" @@ -134,7 +136,6 @@ def test_save_load(moon_dataset): assert embedding is not None assert embedding.shape == (moon_dataset.shape[0], 2) - # if platform.system() != "Windows": # Portable tempfile model_path = tempfile.mkdtemp(suffix="_umap_model")