Read .gospel files in qcheck-stm plugin

n-osborne · Jul 2, 2024 · 1920567 · 1920567
1 parent e9e8f0e
commit 1920567
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/plugins/qcheck-stm/src/config.ml b/plugins/qcheck-stm/src/config.ml
@@ -221,7 +221,12 @@ let init gospel config_module =
   let open Reserr in
   try
     let module_name = Utils.module_name_of_path gospel
-    and env, sigs = Utils.type_check [] gospel in
+    and env, sigs =
+      match Filename.extension gospel with
+      | ".mli" -> Utils.type_check [] gospel
+      | ".gospel" -> Utils.read_gospel_file gospel
+      | _ -> invalid_arg "init"
+    in
     assert (List.length env = 1);
     let namespace = List.hd env in
     let context = Context.init module_name namespace in