fix arbitrary Termprim

AmpersandTarski · Sep 17, 2023 · d5d465c · d5d465c
1 parent 2e3c467
commit d5d465c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/Ampersand/Test/Parser/ArbitraryTree.hs b/src/Ampersand/Test/Parser/ArbitraryTree.hs
@@ -330,12 +330,15 @@ instance Arbitrary TermPrim where
   arbitrary =
     oneof
       [ PI <$> arbitrary,
-        Pid <$> arbitrary <*> arbitrary,
-        Patm <$> arbitrary <*> arbitrary <*> arbitrary,
+        Pid <$> arbitrary <*> arbitrary `suchThat` noLabel,
+        Patm <$> arbitrary <*> arbitrary <*> arbitrary `suchThat` noMaybeLabel,
         PVee <$> arbitrary,
-        Pfull <$> arbitrary <*> arbitrary <*> arbitrary,
+        Pfull <$> arbitrary <*> arbitrary `suchThat` noLabel <*> arbitrary `suchThat` noLabel,
         PNamedR <$> arbitrary
       ]
+    where
+      noMaybeLabel :: Maybe P_Concept -> Bool
+      noMaybeLabel = maybe True noLabel
 
 instance Arbitrary (PairView (Term TermPrim)) where
   arbitrary = PairView <$> arbitrary