diff options
Diffstat (limited to 'src/Test')
-rw-r--r-- | src/Test/VeriFuzz/CodeGen.hs | 5 | ||||
-rw-r--r-- | src/Test/VeriFuzz/VerilogAST.hs | 2 |
2 files changed, 5 insertions, 2 deletions
diff --git a/src/Test/VeriFuzz/CodeGen.hs b/src/Test/VeriFuzz/CodeGen.hs index eb99d6b..85f1d1c 100644 --- a/src/Test/VeriFuzz/CodeGen.hs +++ b/src/Test/VeriFuzz/CodeGen.hs @@ -58,9 +58,12 @@ genExpr _ = "TODO" genPrimary :: Primary -> Text genPrimary (PrimNum num) = - sh (num ^. numSize) <> "'d" <> sh (num ^. numVal) + neg <> sh (num ^. numSize) <> "'d" <> (sh . abs) n where sh = T.pack . show + abs x = if x <= 0 then -x else x + n = num ^. numVal + neg = if n <= 0 then "-" else "" genPrimary (PrimId ident) = ident ^. getIdentifier genBinaryOperator :: BinaryOperator -> Text diff --git a/src/Test/VeriFuzz/VerilogAST.hs b/src/Test/VeriFuzz/VerilogAST.hs index 48d01a5..41c855a 100644 --- a/src/Test/VeriFuzz/VerilogAST.hs +++ b/src/Test/VeriFuzz/VerilogAST.hs @@ -77,7 +77,7 @@ instance QC.Arbitrary Identifier where (QC.shuffle (['a'..'z'] <> ['A'..'Z']) >>= QC.sublistOf) instance QC.Arbitrary Number where - arbitrary = Number <$> QC.arbitrary <*> QC.arbitrary + arbitrary = Number <$> (suchThat QC.arbitrary (>=0)) <*> QC.arbitrary instance QC.Arbitrary BinaryOperator where arbitrary = QC.elements [BinAnd, BinOr, BinXor] |