Update tests

Author: Marius1311

tests/test_cellmapper.py

@@ -39,10 +39,10 @@ def test_expression_transfer_layers(self, cmap, layer_key):
         assert cmap.query_imputed.X.shape[0] == cmap.query.n_obs
 
     @pytest.mark.parametrize("eval_layer", ["X", "counts"])
-    @pytest.mark.parametrize("corr_method", ["pearson", "spearman"])
-    def test_evaluate_expression_transfer_layers_and_methods(self, cmap, eval_layer, corr_method):
+    @pytest.mark.parametrize("method", ["pearson", "spearman", "js", "rmse"])
+    def test_evaluate_expression_transfer_layers_and_methods(self, cmap, eval_layer, method):
         cmap.transfer_expression(layer_key="X")
-        cmap.evaluate_expression_transfer(layer_key=eval_layer, method=corr_method)
+        cmap.evaluate_expression_transfer(layer_key=eval_layer, method=method)
         metrics = cmap.expression_transfer_metrics
-        assert metrics["method"] == corr_method
+        assert metrics["method"] == method
         assert metrics["n_valid_genes"] > 0