Add frequency module in atomic config file

Author: n-osborne

plugins/qcheck-stm/test/atomic_config.ml

@@ -7,6 +7,10 @@ module Gen = struct
   let int = small_int
 end
 
+module Frequencies_dom1 = struct
+  let get = 2
+end
+
 let arb_cmd_dom0 _ =
   let open QCheck in
   make ~print:show_cmd
@@ -24,5 +28,3 @@ let arb_cmd_dom0 _ =
          (1, pure Incr);
          (1, pure Decr);
        ])
-
-let arb_cmd_dom1 = arb_cmd_dom0

plugins/qcheck-stm/test/atomic_tests.expected.ml

@@ -116,7 +116,22 @@ module Spec =
                (1, ((pure (fun n -> Fetch_and_add n)) <*> int));
                (1, (pure Incr));
                (1, (pure Decr))])
-    let arb_cmd_dom1 = arb_cmd_dom0
+    let arb_cmd_dom1 _ =
+      let open QCheck in
+        make ~print:show_cmd
+          (let open Gen in
+             frequency
+               [(1, ((pure (fun v -> Make v)) <*> small_signed_int));
+               (2, (pure Get));
+               (1, ((pure (fun v_1 -> Set v_1)) <*> int));
+               (1, ((pure (fun v_2 -> Exchange v_2)) <*> int));
+               (1,
+                 (((pure (fun seen v_3 -> Compare_and_set (seen, v_3))) <*>
+                     int)
+                    <*> int));
+               (1, ((pure (fun n -> Fetch_and_add n)) <*> int));
+               (1, (pure Incr));
+               (1, (pure Decr))])
     let next_state cmd__002_ state__003_ =
       match cmd__002_ with
       | Make v ->