Skip to content

Commit 8381bb7

Browse files
authored
testing exclusion of observed data from posterior median too
1 parent 3f7a580 commit 8381bb7

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

tests/test_cell2location.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717

1818
def export_posterior(model, dataset):
1919
dataset = model.export_posterior(dataset, use_quantiles=True, add_to_obsm=["q50", "q001"]) # quantile 0.50
20+
assert "data_target" not in dataset.uns["mod"]["post_sample_q50"].keys()
2021
dataset = model.export_posterior(
2122
dataset, use_quantiles=True, add_to_obsm=["q50"], sample_kwargs={"batch_size": 10}
2223
) # quantile 0.50
@@ -26,6 +27,7 @@ def export_posterior(model, dataset):
2627

2728
def export_posterior_sc(model, dataset):
2829
dataset = model.export_posterior(dataset, use_quantiles=True, add_to_varm=["q50", "q001"]) # quantile 0.50
30+
assert "data_target" not in dataset.uns["mod"]["post_sample_q50"].keys()
2931
dataset = model.export_posterior(
3032
dataset, use_quantiles=True, add_to_varm=["q50"], sample_kwargs={"batch_size": 10}
3133
) # quantile 0.50

0 commit comments

Comments
 (0)