test: simplify variablecollection generation to speed up generation s…

…ome more – allow shrinking of ivs, dvs and covariates together
AutoResearch · Nov 22, 2023 · 901253f · 901253f
1 parent 50dd66a
commit 901253f
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tests/test_strategies.py b/tests/test_strategies.py
@@ -128,9 +128,13 @@ def variablecollection_strategy(
     if num_variables is not None:
         n_ivs, n_dvs, n_covariates = num_variables
     else:  # num_variables is None
-        n_ivs = draw(st.integers(min_value=1, max_value=max_length))
-        n_dvs = draw(st.integers(min_value=1, max_value=max_length))
-        n_covariates = draw(st.integers(min_value=0, max_value=max_length))
+        n_ivs, n_dvs, n_covariates = draw(
+            st.tuples(
+                st.integers(min_value=1, max_value=max_length),
+                st.integers(min_value=1, max_value=max_length),
+                st.integers(min_value=0, max_value=max_length),
+            )
+        )
 
     n_variables = n_ivs + n_dvs + n_covariates