| """Tests for PT state discovery.""" | |
| import numpy as np | |
| import pytest | |
| def test_pt_states(analyzed_adata): | |
| assert "pt_state" in analyzed_adata.obs.columns | |
| assert "X_gamma_pca" in analyzed_adata.obsm | |
| assert "X_gamma_umap" in analyzed_adata.obsm | |
| assert analyzed_adata.obs["pt_state"].dtype.name == "category" | |
| n_states = analyzed_adata.obs["pt_state"].nunique() | |
| assert n_states >= 1 | |
| def test_rank_pt_genes(analyzed_adata): | |
| import scptr | |
| result = scptr.tl.rank_pt_genes(analyzed_adata, groupby="pt_state") | |
| assert len(result) > 0 | |
| assert "names" in result.columns or "group" in result.columns | |
| assert "rank_pt_genes" in analyzed_adata.uns["scptr"] | |