aboutsummaryrefslogtreecommitdiffstats
path: root/test/Unit.hs
blob: 9cc75fb95d4000747c7d435c4c1850db351f818f (plain)
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
module Unit
  ( unitTests,
  )
where

import Control.Lens
import Data.List.NonEmpty (NonEmpty (..))
import Parser (parseUnitTests)
import Reduce (reduceUnitTests)
import Test.Tasty
import Test.Tasty.HUnit
import Verismith

unitTests :: TestTree
unitTests =
  testGroup
    "Unit tests"
    [ testCase "Transformation of AST" $
        assertEqual
          "Successful transformation"
          transformExpectedResult
          (transform trans transformTestData),
      parseUnitTests,
      reduceUnitTests
    ]

transformTestData :: Expr
transformTestData =
  BinOp
    ( BinOp
        (BinOp (Id "id1") BinAnd (Id "id2"))
        BinAnd
        (BinOp (Id "id1") BinAnd (Id "id2"))
    )
    BinAnd
    ( BinOp
        ( BinOp
            (BinOp (Id "id1") BinAnd (Id "id2"))
            BinAnd
            ( BinOp
                (Id "id1")
                BinAnd
                ( BinOp
                    (BinOp (Id "id1") BinAnd (Id "id2"))
                    BinAnd
                    (BinOp (Id "id1") BinAnd (Id "id2"))
                )
            )
        )
        BinOr
        ( Concat $
            ( Concat $
                (Concat $ (Id "id1") :| [Id "id2", Id "id2"])
                  :| [ Id "id2",
                       Id "id2",
                       ( Concat $
                           (Id "id2")
                             :| [Id "id2", (Concat $ Id "id1" :| [Id "id2"])]
                       ),
                       Id "id2"
                     ]
            )
              :| [Id "id1", Id "id2"]
        )
    )

transformExpectedResult :: Expr
transformExpectedResult =
  BinOp
    ( BinOp
        (BinOp (Id "id1") BinAnd (Id "Replaced"))
        BinAnd
        (BinOp (Id "id1") BinAnd (Id "Replaced"))
    )
    BinAnd
    ( BinOp
        ( BinOp
            (BinOp (Id "id1") BinAnd (Id "Replaced"))
            BinAnd
            ( BinOp
                (Id "id1")
                BinAnd
                ( BinOp
                    (BinOp (Id "id1") BinAnd (Id "Replaced"))
                    BinAnd
                    (BinOp (Id "id1") BinAnd (Id "Replaced"))
                )
            )
        )
        BinOr
        ( Concat $
            ( Concat $
                (Concat $ (Id "id1") :| [Id "Replaced", Id "Replaced"])
                  :| [ Id "Replaced",
                       Id "Replaced",
                       Concat $
                         Id "Replaced"
                           :| [Id "Replaced", Concat $ Id "id1" :| [Id "Replaced"]],
                       Id "Replaced"
                     ]
            )
              :| [Id "id1", Id "Replaced"]
        )
    )

trans :: Expr -> Expr
trans e = case e of
  Id i -> if i == Identifier "id2" then Id $ Identifier "Replaced" else Id i
  _ -> e