fix again

yet-another-ai · Dec 12, 2023 · 6b2cf67 · 6b2cf67
1 parent b334693
commit 6b2cf67
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/lib/prolog_mqi/prolog_mqi.rb b/lib/prolog_mqi/prolog_mqi.rb
@@ -51,7 +51,7 @@ def query(value)
       read
     end
 
-    def query_async(value, find_all=true)
+    def query_async(value, find_all: true)
       timeout_str = @timeout.nil? ? '_' : @timeout.to_s
       find_all_str = find_all ? 'true' : 'false'
       write("run_async(#{value}, #{timeout_str}, #{find_all_str})")
@@ -69,6 +69,13 @@ def query_async_result
       read
     end
 
+    def halt_server
+      write('halt')
+      read
+    end
+
+    private
+
     def read
       bytesize = @socket.gets.chomp(".\n").to_i
       result = @socket.read(bytesize)
@@ -85,6 +92,8 @@ def write(value)
     end
 
     def stop
+      halt_server
+    ensure
       @socket.close
       @stdin.close
       @stdout.close

diff --git a/test/bench_prolog_mqi.rb b/test/bench_prolog_mqi.rb
@@ -45,7 +45,7 @@ def bench_revert_facts_instead_of_restart
 
     puts "bench_revert_facts_instead_of_restart\t#{time_no_restart.real}\t#{time_restart.real}"
 
-    assert time_no_restart.real < time_restart.real
+    assert_operator time_no_restart.real, :<, time_restart.real
   end
   # rubocop:enable Metrics/AbcSize
 end