test distance correlation

Former-commit-id: 8e26833

Author: scottgigante-immunai

test/test_task_dimensionality_reduction.py

@@ -1,5 +1,6 @@
 """Specific tests for the dimensionality_reduction task"""
 import openproblems
+import parameterized
 import utils.docker
 import utils.git
 
@@ -54,6 +55,26 @@ def test_density_preservation_matches_densmap():
     np.testing.assert_allclose(expected, actual, rtol=1e-3)
 
 
+@parameterized.parameterized.expand(
+    [(200,), (1000,)],
+    name_func=utils.name.name_test,
+)
+def test_distance_correlation_with_svd(n_svd):
+    import numpy as np
+
+    task = openproblems.tasks.dimensionality_reduction
+    metric = openproblems.tasks.dimensionality_reduction.metrics.distance_correlation
+
+    adata = task.api.sample_dataset()
+    adata = task.api.sample_method(adata)
+    adata.obsm["X_emb"] = adata.X.toarray()
+
+    expected = 1
+    actual = metric(adata, n_svd=n_svd)
+
+    np.testing.assert_allclose(expected, actual, rtol=1e-3)
+
+
 def test_density_preservation_perfect():
     import numpy as np