diff --git a/tests/evals/data/testdata-2d-nan.pkl.gz b/tests/evals/data/testdata-2d-nan.pkl.gz new file mode 100644 index 00000000..abd1337e Binary files /dev/null and b/tests/evals/data/testdata-2d-nan.pkl.gz differ diff --git a/tests/evals/data/testdata-2d.pkl.gz b/tests/evals/data/testdata-2d.pkl.gz new file mode 100644 index 00000000..c04edb5b Binary files /dev/null and b/tests/evals/data/testdata-2d.pkl.gz differ diff --git a/tests/evals/test_metrics.py b/tests/evals/test_metrics.py index 613e773b..c6bed918 100644 --- a/tests/evals/test_metrics.py +++ b/tests/evals/test_metrics.py @@ -69,7 +69,7 @@ def test_pattern_correlation(self): self.assertEqual(pattern_correlation(x, y).shape, (10,)) def test_2d(self): - with open('data/testdata-2d.pkl.gz', 'rb') as f: + with open('tests/evals/data/testdata-2d.pkl.gz', 'rb') as f: d = pickle.load(f) self.assertTrue(np.array_equal( profile_correlation(d['x'], d['y']), @@ -85,7 +85,7 @@ def test_2d(self): )) def test_2d_nan(self): - with open('data/testdata-2d-nan.pkl.gz', 'rb') as f: + with open('tests/evals/data/testdata-2d-nan.pkl.gz', 'rb') as f: d = pickle.load(f) # self.assertTrue(np.array_equal( # profile_correlation(d['x'], d['y']),