prop_filter for CustomType

CQCL · Dec 23, 2024 · 86df78d · 86df78d
1 parent f7cca46
commit 86df78d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/hugr-core/src/types.rs b/hugr-core/src/types.rs
@@ -802,7 +802,11 @@ pub(crate) mod test {
             ) -> impl Strategy<Value = (Type, Vec<TypeParam>)> + Clone {
                 prop_oneof![
                     // no Alias
-                    MakeCustomType.with_env(vars.clone(), reg.clone()),
+                    MakeCustomType
+                        .with_env(vars.clone(), reg.clone())
+                        .prop_filter("Must fit TypeBound", move |(ct, _)| self
+                            .0
+                            .contains(ct.least_upper_bound())),
                     MakeFuncType.with_env(vars.clone(), reg.clone()),
                     make_type_var(self.0.into(), vars.clone()).prop_map(|(ta, vars)| match ta {
                         TypeArg::Type { ty } => (ty, vars),