Add union test

Signed-off-by: Ana Pantilie <ana.pantilie95@gmail.com>
IntersectMBO · ana-pantilie · May 21, 2024 · Apr 29, 2024 · Apr 29, 2024 · Apr 29, 2024
commit 7d96dc408a07fa4c37d03b5ba733d5918fbd6e65
diff --git a/plutus-tx-plugin/test/AssocMap/Spec.hs b/plutus-tx-plugin/test/AssocMap/Spec.hs
@@ -32,7 +32,7 @@ import PlutusTx.Prelude qualified as PlutusTx
 import PlutusTx.Show qualified as PlutusTx
 import PlutusTx.Test
 import PlutusTx.TH (compile)
-import PlutusTx.These (These (..))
+import PlutusTx.These (These (..), these)
 import Test.Tasty (TestTree, testGroup)
 import Test.Tasty.Hedgehog (testProperty)
 
@@ -48,6 +48,10 @@ goldenTests =
     , goldenUPlcReadable "map2" map2
     , goldenEvalCekCatch "map2" $ [map2 `unsafeApplyCode` (liftCodeDef 100)]
     , goldenBudget "map2-budget" $ map2 `unsafeApplyCode` (liftCodeDef 100)
+    , goldenPirReadable "map3" map2
+    , goldenUPlcReadable "map3" map2
+    , goldenEvalCekCatch "map3" $ [map2 `unsafeApplyCode` (liftCodeDef 100)]
+    , goldenBudget "map3-budget" $ map2 `unsafeApplyCode` (liftCodeDef 100)
     ]
 
 propertyTests :: TestTree
@@ -127,6 +131,32 @@ map2 =
         ||]
     )
 
+map3 :: CompiledCode (Integer -> [(Integer, PlutusTx.BuiltinString)])
+map3 =
+  $$( compile
+        [||
+        \n ->
+          let m1 =
+                Data.AssocMap.unsafeFromList
+                  [ (n PlutusTx.+ 1, "one")
+                  , (n PlutusTx.+ 2, "two")
+                  , (n PlutusTx.+ 3, "three")
+                  , (n PlutusTx.+ 4, "four")
+                  , (n PlutusTx.+ 5, "five")
+                  ]
+              m2 =
+                Data.AssocMap.unsafeFromList
+                  [ (n PlutusTx.+ 3, "THREE")
+                  , (n PlutusTx.+ 4, "FOUR")
+                  , (n PlutusTx.+ 6, "SIX")
+                  , (n PlutusTx.+ 7, "SEVEN")
+                  ]
+              m = Data.AssocMap.union m1 m2
+              f = these id id PlutusTx.appendByteString
+           in PlutusTx.fmap (\(k, v) -> (k, PlutusTx.decodeUtf8 (f v))) (Data.AssocMap.toList m)
+        ||]
+    )
+
 -- | The semantics of PlutusTx maps and their operations.
 -- The 'PlutusTx' implementations maps ('Data.AssocMap.Map' and 'AssocMap.Map')
 -- are checked against the semantics to ensure correctness.

diff --git a/plutus-tx-plugin/test/Budget/9.6/map3-budget.budget.golden b/plutus-tx-plugin/test/Budget/9.6/map3-budget.budget.golden
@@ -0,0 +1,2 @@
+({cpu: 160633417
+| mem: 416622})
diff --git a/plutus-tx-plugin/test/Budget/9.6/map3.eval.golden b/plutus-tx-plugin/test/Budget/9.6/map3.eval.golden
@@ -0,0 +1,27 @@
+(constr
+  1
+  (constr 0 (con integer 105) (con string "five"))
+  (constr
+    1
+    (constr 0 (con integer 104) (con string "fourFOUR"))
+    (constr
+      1
+      (constr 0 (con integer 103) (con string "threeTHREE"))
+      (constr
+        1
+        (constr 0 (con integer 102) (con string "two"))
+        (constr
+          1
+          (constr 0 (con integer 101) (con string "one"))
+          (constr
+            1
+            (constr 0 (con integer 106) (con string "SIX"))
+            (constr
+              1 (constr 0 (con integer 107) (con string "SEVEN")) (constr 0)
+            )
+          )
+        )
+      )
+    )
+  )
+)