pydanting

Author: Jon Brenas

tests/anoph/test_pca.py

@@ -344,11 +344,11 @@ def test_pca_fit_exclude_samples(fixture, api: AnophelesPca):
 def test_pca_cohort_downsampling(fixture, api: AnophelesPca):
     # Parameters for selecting input data.
     all_sample_sets = api.sample_sets()["sample_set"].to_list()
-    sample_sets = np.random.choice(all_sample_sets, size=2)
+    sample_sets = np.random.choice(all_sample_sets, size=2, replace=False).tolist()
     data_params = dict(
-        region=np.random.choice(api.contigs),
+        region=str(np.random.choice(api.contigs)),
         sample_sets=sample_sets,
-        site_mask=np.random.choice(api.site_mask_ids),
+        site_mask=np.random.choice(list(api.site_mask_ids) + [None]),
     )
 
     # Test cohort downsampling.