aboutsummaryrefslogtreecommitdiffstats
path: root/src/VeriFuzz/Internal/Gen.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/VeriFuzz/Internal/Gen.hs')
-rw-r--r--src/VeriFuzz/Internal/Gen.hs21
1 files changed, 14 insertions, 7 deletions
diff --git a/src/VeriFuzz/Internal/Gen.hs b/src/VeriFuzz/Internal/Gen.hs
index be275dd..6e44524 100644
--- a/src/VeriFuzz/Internal/Gen.hs
+++ b/src/VeriFuzz/Internal/Gen.hs
@@ -12,18 +12,25 @@ Internal helpers for generation.
module VeriFuzz.Internal.Gen where
-import Data.Graph.Inductive (Graph, Node)
-import qualified Data.Graph.Inductive as G
-import qualified Data.Text as T
+import Data.Graph.Inductive ( Graph
+ , Node
+ )
+import qualified Data.Graph.Inductive as G
+import qualified Data.Text as T
fromNode :: Int -> T.Text
fromNode node = T.pack $ "w" <> show node
filterGr :: (Graph gr) => gr n e -> (Node -> Bool) -> [Node]
-filterGr graph f =
- filter f $ G.nodes graph
-
-only :: (Graph gr) => gr n e -> (gr n e -> Node -> Int) -> (gr n e -> Node -> Int) -> Node -> Bool
+filterGr graph f = filter f $ G.nodes graph
+
+only
+ :: (Graph gr)
+ => gr n e
+ -> (gr n e -> Node -> Int)
+ -> (gr n e -> Node -> Int)
+ -> Node
+ -> Bool
only graph fun1 fun2 n = fun1 graph n == 0 && fun2 graph n /= 0
inputs :: (Graph gr) => gr n e -> [Node]