1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
{-# OPTIONS_GHC -fno-warn-orphans #-}
module Futhark.BenchTests (tests) where
import Data.Map qualified as M
import Data.Text qualified as T
import Futhark.Bench
import Futhark.ProfileTests ()
import Test.Tasty
import Test.Tasty.QuickCheck
instance Arbitrary RunResult where
arbitrary = RunResult . getPositive <$> arbitrary
printable :: Gen String
printable = getASCIIString <$> arbitrary
instance Arbitrary DataResult where
arbitrary =
DataResult
<$> (T.pack <$> printable)
<*> oneof
[ Left <$> arbText,
Right
<$> ( Result
<$> arbitrary
<*> arbMap
<*> oneof [pure Nothing, Just <$> arbText]
<*> arbitrary
)
]
where
arbText = T.pack <$> printable
arbMap = M.fromList <$> listOf ((,) <$> arbText <*> arbitrary)
-- XXX: we restrict this generator to single datasets to we don't have
-- to worry about duplicates.
instance Arbitrary BenchResult where
arbitrary = BenchResult <$> printable <*> (pure <$> arbitrary)
encodeDecodeJSON :: TestTree
encodeDecodeJSON = testProperty "encoding and decoding are inverse" prop
where
prop :: BenchResult -> Bool
prop brs = decodeBenchResults (encodeBenchResults [brs]) == Right [brs]
tests :: TestTree
tests = testGroup "Futhark.BenchTests" [encodeDecodeJSON]
|
