@@ -111,15 +111,14 @@ def test_plot_label_categorical_color(self, sdata_blobs: SpatialData, label: str
111
111
self ._make_tablemodel_with_categorical_labels (sdata_blobs , label )
112
112
113
113
def _make_tablemodel_with_categorical_labels (self , sdata_blobs , label ):
114
+
114
115
n_obs = max (get_element_instances (sdata_blobs [label ]))
115
- adata = AnnData (
116
- RNG .normal (size = (n_obs , 10 )),
117
- obs = pd .DataFrame (RNG .normal (size = (n_obs , 3 )), columns = ["a" , "b" , "c" ]),
118
- )
119
- adata .obs ["instance_id" ] = np .arange (adata .n_obs )
120
- adata .obs ["category" ] = RNG .choice (["a" , "b" , "c" ], size = adata .n_obs )
121
- adata .obs ["category" ][:3 ] = ["a" , "b" , "c" ]
122
- adata .obs ["instance_id" ] = list (range (adata .n_obs ))
116
+ vals = np .arange (n_obs )
117
+ obs = pd .DataFrame ({"a" : vals , "b" : vals + 0.3 , "c" : vals + 0.7 })
118
+
119
+ adata = AnnData (vals .reshape (- 1 , 1 ), obs = obs )
120
+ adata .obs ["instance_id" ] = vals
121
+ adata .obs ["category" ] = list (["a" , "b" , "c" ] * ((n_obs // 3 ) + 1 ))[:n_obs ]
123
122
adata .obs ["region" ] = label
124
123
table = TableModel .parse (
125
124
adata = adata ,
0 commit comments