Bug fix

scverse · Mar 28, 2020 · 56bcebc · 56bcebc
1 parent f0fabe8
commit 56bcebc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_plotting.py b/tests/test_plotting.py
@@ -41,8 +41,8 @@ def test_spectratype(adata_tra):
     p = pl.spectratype(adata_tra, target_col="sample")
     assert isinstance(p, plt.Axes)
 
-def test_vdj_usage(adata_tra):
-    p = pl.vdj_usage(adata_tra, cell_weights="sample")
+def test_vdj_usage(adata_vdj):
+    p = pl.vdj_usage(adata_vdj, cell_weights="sample")
     assert isinstance(p, plt.Axes)
 
 def test_clonotype_network(adata_clonotype_network):

diff --git a/tests/test_tools.py b/tests/test_tools.py
@@ -218,7 +218,7 @@ def test_spectratype(adata_tra):
 def test_vdj_usage(adata_vdj):
     # Check return formats
     df = st.tl.vdj_usage(adata_vdj, cell_weights="sample", as_dict=False)
-    assert isinstance(df, AnnData)
+    assert isinstance(df, pd.DataFrame)
     df = st.tl.vdj_usage(adata_vdj, cell_weights="sample", as_dict=True)
     assert isinstance(df, dict)