Enforce stable cluster order in notebook test (scverse#2650)

HaidYi · Sep 4, 2023 · c87f1d6 · c87f1d6
1 parent 1efc3a7
commit c87f1d6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/scanpy/tests/conftest.py b/scanpy/tests/conftest.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import sys
 from pathlib import Path

diff --git a/scanpy/tests/notebooks/test_pbmc3k.py b/scanpy/tests/notebooks/test_pbmc3k.py
@@ -129,6 +129,11 @@ def test_pbmc3k(image_comparer):
     # sc.pl.rank_genes_groups_violin(adata, groups='0', n_genes=8)
     # save_and_compare_images('rank_genes_groups_4')
 
+    if adata[adata.obs['leiden'] == '4', 'CST3'].X.mean() < 1:
+        (  # switch clusters
+            adata.obs['leiden'][adata.obs['leiden'] == '4'],
+            adata.obs['leiden'][adata.obs['leiden'] == '5'],
+        ) = ('5', '4')
     new_cluster_names = [
         'CD4 T cells',
         'CD14+ Monocytes',