z3 0.0.20220630 → 0.0.20221020

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a9289c94f0a177c8e59d2c18c65775344414bb3558151ac58fc02e932b3a7eb1
-  data.tar.gz: 3171e750dc3a36d1534ffc0c069274ff8bc7bc4d59d15e6792add5c4ab5f90b6
+  metadata.gz: abba86119ff297f1b65693b634c5f33b11e53ea24f76b5b851d11918bf061694
+  data.tar.gz: bf2253285a83d24d1e7a5264d839506f65b32a45c27203dcf12202daab0c527a
 SHA512:
-  metadata.gz: d19511280987279176db440a5c5f3764d9029b49c74185c32cbf3dd01e6f02bcefa92b5800b24d4e872f052cb40a8daddf756783c81f3fdb63055d1be950b119
-  data.tar.gz: '09eedf827d3dfae3c56fd6d1c165160cdb5e3aeab0e2faae429dd81da91b7eaafd6640a6f28ed2a1b45e8a3a3c3e12a03d38327e86ac497b69e974e5f431cee6'
+  metadata.gz: d63a2a62b5f8103eca1c0ae9429a5538cb5a0add23ca14fdceb1759f821d5a0f8ddeb22c06efad3780b5498d8c6ff0fea4f9d0c86c3d1e1dbfd443265f3b4f4a
+  data.tar.gz: 7cd9d08794c1fa4cb2c47575c3977d35a781874c5f27f03ee31eb251e153062f91e8bd38236da887cc1a1fb6e401250038e98cd44d63197e120d84cd8171ef1f

data/lib/z3/expr/array_expr.rb

@@ -1,5 +1,25 @@
 module Z3
   class ArrayExpr < Expr
     public_class_method :new
+
+    def key_sort
+      sort.key_sort
+    end
+
+    def value_sort
+      sort.value_sort
+    end
+
+    def store(key, value)
+      sort.new LowLevel.mk_store(self, key_sort.cast(key), value_sort.cast(value))
+    end
+
+    def select(key)
+      sort.value_sort.new LowLevel.mk_select(self, key_sort.cast(key))
+    end
+
+    def [](key)
+      select(key)
+    end
   end
 end

data/lib/z3/interface.rb

@@ -72,6 +72,10 @@ module Z3
     LowLevel.get_version.join(".")
   end
 
+  def version_at_least?(a, b=0, c=0, d=0)
+    (LowLevel.get_version <=> [a, b, c, d]) >= 0
+  end
+
   def set_param(k,v)
     LowLevel.global_param_set(k,v)
   end

data/lib/z3/optimize.rb

@@ -23,9 +23,9 @@ module Z3
       LowLevel.optimize_assert(self, ast)
     end
 
-    def assert_soft(ast)
+    def assert_soft(ast, weight = "1", id = nil)
       reset_model!
-      LowLevel.optimize_assert_soft(self, ast)
+      LowLevel.optimize_assert_soft(self, ast, weight, id)
     end
 
     def check(*args)

data/spec/array_expr_spec.rb

@@ -4,7 +4,9 @@ module Z3
     let(:a) { sort.var("a") }
     let(:b) { sort.var("b") }
     let(:c) { sort.var("c") }
-    let(:x) { Z3::Bool("x") }
+    let(:x) { Z3::Int("x") }
+    let(:y) { Z3::Int("y") }
+    let(:z) { Z3::Int("z") }
 
     # TODO: Formatting is dreadful
     it "== and !=" do
@@ -14,5 +16,35 @@ module Z3
         c => "const(0)",
       )
     end
+
+    it "select" do
+      expect([a.select(10) == 20]).to have_solution(
+        a => "const(20)",
+      )
+      expect([a[10] == 20]).to have_solution(
+        a => "const(20)",
+      )
+      expect([a[x] == 10, a[y] == 20]).to have_solution(
+        a => "store(const(10), 3, 20)",
+        x => "2",
+        y => "3",
+      )
+    end
+
+    it "store" do
+      expect([a == b.store(10, 20), x == a.select(10)]).to have_solution(
+        x => 20,
+      )
+      expect([a == b.store(10, 20), x == a[10]]).to have_solution(
+        x => 20,
+      )
+      expect([a == b.store(10, 20), x == a[y], y == 10]).to have_solution(
+        x => 20,
+      )
+      expect([a == b.store(10, 20), x == a[y], x == 20]).to have_solution(
+        x => 20,
+        y => 10,
+      )
+    end
   end
 end

data/spec/array_sort_spec.rb

@@ -1,7 +1,7 @@
 module Z3
   describe ArraySort do
-    let(:int_int_array)   { ArraySort.new(IntSort.new, IntSort.new) }
-    let(:int_real_array)  { ArraySort.new(IntSort.new, RealSort.new) }
+    let(:int_int_array)  { ArraySort.new(IntSort.new, IntSort.new) }
+    let(:int_real_array) { ArraySort.new(IntSort.new, RealSort.new) }
 
     it "can instantiate variables" do
       expect(int_int_array.var("a").inspect).to eq("Array(Int, Int)<a>")

data/spec/float_expr_spec.rb

@@ -102,8 +102,9 @@ module Z3
       expect(positive_infinity.to_s).to eq("+oo")
       expect(negative_infinity.to_s).to eq("-oo")
       expect(nan.to_s).to eq("NaN")
+
       # Denormals changed
-      if Z3.version >= '4.5'
+      if Z3.version_at_least?(4, 6)
         expect(float_double.from_const(1234 * 0.5**1040).to_s).to eq("0.00470733642578125B-1022")
         expect(float_single.from_const(1234 * 0.5**136).to_s).to eq("1.205078125B-126")
         # This is what we get, all of these are wrong, by a lot:

data/spec/integration/abc_path_spec.rb

@@ -1,21 +1,5 @@
-# There are multiple solutions, so this test is nondeterministic
-# This is what Z3 4.8.13 returns
 describe "ABC Path" do
   it do
-    expect("abc_path").to have_output <<'EOF'
-C O R D F B U
-
-N o-n l-k-j L
-  | |/    |
-G p m g-f i I
-  |    x /
-Q q-r e h a H
-   /  |   |
-V s v d-c-b S
-   x \
-T u-t w-x-y X
-
-J P M W K Y E
-EOF
+    expect("abc_path").to have_output_matching_saved_example
   end
 end

data/spec/integration/examples/abc_path-1.txt

@@ -0,0 +1,13 @@
+C O R D F B U
+
+N o-n l-k-j L
+  | |/    |
+G p m g-f i I
+  |    x /
+Q q-r e h a H
+   /  |   |
+V s v d-c-b S
+   x \
+T u-t w-x-y X
+
+J P M W K Y E

data/spec/integration/examples/abc_path-2.txt

@@ -0,0 +1,13 @@
+C O R D F B U
+
+N o-n l-k-j L
+  | |/    |
+G p m g-f i I
+  |    x /
+Q q-r e h b H
+   /  |  /|
+V s v d-c a S
+   x \
+T u-t w-x-y X
+
+J P M W K Y E

data/spec/integration/examples/futoshiki-1.txt

@@ -0,0 +1,17 @@
+2 5 3 4 1 7 6<9 8
+
+1 9 7 2 8 5 3<4<6
+  _
+7<8 2 5>3>1 4<6 9
+^
+8 1<5<9 7 6 2 3 4
+            ^
+4 7>6>1 9 2 5 8 3
+^         ^     ^
+6 4 1 8 2 3 9 5 7
+  _   _ ^   _ _
+9 3 4 6>5 8 7 1 2
+_
+5 2 9>3 6>4 8>7 1
+              _
+3 6 8>7>4 9 1<2 5

data/spec/integration/examples/futoshiki-2.txt

@@ -0,0 +1,17 @@
+2 5 3 4 1 7 6<9 8
+
+1 9 7 2 8 5 3<4<6
+  _
+7<8 2 5>3>1 4<6 9
+^
+8 3<5<9 7 6 2 1 4
+            ^
+4 7>6>1 9 2 5 8 3
+^         ^     ^
+6 4 1 8 2 3 9 5 7
+  _   _ ^   _ _
+9 1 4 6>5 8 7 3 2
+_
+5 2 9>3 6>4 8>7 1
+              _
+3 6 8>7>4 9 1<2 5

data/spec/integration/examples/knights_puzzle-1.txt

@@ -0,0 +1,90 @@
+Solved
+State 0:
+bbb.
+xbxw
+..ww
+x.xw
+w: 2,2 -> 3,0
+
+State 1:
+bbbw
+xbxw
+...w
+x.xw
+b: 1,0 -> 2,2
+
+State 2:
+b.bw
+xbxw
+..bw
+x.xw
+w: 3,1 -> 1,0
+
+State 3:
+bwbw
+xbx.
+..bw
+x.xw
+b: 0,0 -> 1,2
+
+State 4:
+.wbw
+xbx.
+.bbw
+x.xw
+b: 1,2 -> 3,1
+
+State 5:
+.wbw
+xbxb
+..bw
+x.xw
+w: 3,3 -> 1,2
+
+State 6:
+.wbw
+xbxb
+.wbw
+x.x.
+w: 1,2 -> 0,0
+
+State 7:
+wwbw
+xbxb
+..bw
+x.x.
+b: 2,0 -> 1,2
+
+State 8:
+ww.w
+xbxb
+.bbw
+x.x.
+w: 3,2 -> 2,0
+
+State 9:
+wwww
+xbxb
+.bb.
+x.x.
+b: 1,1 -> 3,2
+
+State 10:
+wwww
+x.xb
+.bbb
+x.x.
+b: 1,2 -> 3,3
+
+State 11:
+wwww
+x.xb
+..bb
+x.xb
+w: 3,0 -> 1,1
+
+State 12:
+www.
+xwxb
+..bb
+x.xb

data/spec/integration/examples/knights_puzzle-2.txt

@@ -0,0 +1,90 @@
+Solved
+State 0:
+bbb.
+xbxw
+..ww
+x.xw
+w: 2,2 -> 3,0
+
+State 1:
+bbbw
+xbxw
+...w
+x.xw
+b: 1,0 -> 2,2
+
+State 2:
+b.bw
+xbxw
+..bw
+x.xw
+w: 3,1 -> 1,0
+
+State 3:
+bwbw
+xbx.
+..bw
+x.xw
+b: 0,0 -> 1,2
+
+State 4:
+.wbw
+xbx.
+.bbw
+x.xw
+b: 1,2 -> 3,1
+
+State 5:
+.wbw
+xbxb
+..bw
+x.xw
+w: 3,3 -> 1,2
+
+State 6:
+.wbw
+xbxb
+.wbw
+x.x.
+w: 1,2 -> 0,0
+
+State 7:
+wwbw
+xbxb
+..bw
+x.x.
+b: 2,0 -> 1,2
+
+State 8:
+ww.w
+xbxb
+.bbw
+x.x.
+b: 1,2 -> 3,3
+
+State 9:
+ww.w
+xbxb
+..bw
+x.xb
+w: 3,2 -> 2,0
+
+State 10:
+wwww
+xbxb
+..b.
+x.xb
+b: 1,1 -> 3,2
+
+State 11:
+wwww
+x.xb
+..bb
+x.xb
+w: 3,0 -> 1,1
+
+State 12:
+www.
+xwxb
+..bb
+x.xb

data/spec/integration/examples/knights_puzzle-3.txt

@@ -0,0 +1,90 @@
+Solved
+State 0:
+bbb.
+xbxw
+..ww
+x.xw
+w: 2,2 -> 3,0
+
+State 1:
+bbbw
+xbxw
+...w
+x.xw
+b: 1,0 -> 2,2
+
+State 2:
+b.bw
+xbxw
+..bw
+x.xw
+b: 0,0 -> 1,2
+
+State 3:
+..bw
+xbxw
+.bbw
+x.xw
+w: 3,1 -> 1,0
+
+State 4:
+.wbw
+xbx.
+.bbw
+x.xw
+b: 1,2 -> 3,1
+
+State 5:
+.wbw
+xbxb
+..bw
+x.xw
+w: 3,3 -> 1,2
+
+State 6:
+.wbw
+xbxb
+.wbw
+x.x.
+w: 1,2 -> 0,0
+
+State 7:
+wwbw
+xbxb
+..bw
+x.x.
+b: 2,0 -> 1,2
+
+State 8:
+ww.w
+xbxb
+.bbw
+x.x.
+b: 1,2 -> 3,3
+
+State 9:
+ww.w
+xbxb
+..bw
+x.xb
+w: 3,2 -> 2,0
+
+State 10:
+wwww
+xbxb
+..b.
+x.xb
+b: 1,1 -> 3,2
+
+State 11:
+wwww
+x.xb
+..bb
+x.xb
+w: 3,0 -> 1,1
+
+State 12:
+www.
+xwxb
+..bb
+x.xb

data/spec/integration/examples/knights_puzzle-4.txt

@@ -0,0 +1,90 @@
+Solved
+State 0:
+bbb.
+xbxw
+..ww
+x.xw
+w: 3,3 -> 1,2
+
+State 1:
+bbb.
+xbxw
+.www
+x.x.
+b: 1,1 -> 3,0
+
+State 2:
+bbbb
+x.xw
+.www
+x.x.
+w: 3,2 -> 1,1
+
+State 3:
+bbbb
+xwxw
+.ww.
+x.x.
+b: 2,0 -> 3,2
+
+State 4:
+bb.b
+xwxw
+.wwb
+x.x.
+w: 1,2 -> 2,0
+
+State 5:
+bbwb
+xwxw
+..wb
+x.x.
+b: 0,0 -> 1,2
+
+State 6:
+.bwb
+xwxw
+.bwb
+x.x.
+b: 1,2 -> 3,3
+
+State 7:
+.bwb
+xwxw
+..wb
+x.xb
+w: 3,1 -> 1,2
+
+State 8:
+.bwb
+xwx.
+.wwb
+x.xb
+b: 1,0 -> 3,1
+
+State 9:
+..wb
+xwxb
+.wwb
+x.xb
+w: 2,2 -> 1,0
+
+State 10:
+.wwb
+xwxb
+.w.b
+x.xb
+b: 3,0 -> 2,2
+
+State 11:
+.ww.
+xwxb
+.wbb
+x.xb
+w: 1,2 -> 0,0
+
+State 12:
+www.
+xwxb
+..bb
+x.xb

data/spec/integration/futoshiki_spec.rb

@@ -1,23 +1,5 @@
 describe "Futoshiki" do
   it do
-    expect("futoshiki").to have_output <<EOF
-2 5 3 4 1 7 6<9 8
-
-1 9 7 2 8 5 3<4<6
-  _
-7<8 2 5>3>1 4<6 9
-^
-8 1<5<9 7 6 2 3 4
-            ^
-4 7>6>1 9 2 5 8 3
-^         ^     ^
-6 4 1 8 2 3 9 5 7
-  _   _ ^   _ _
-9 3 4 6>5 8 7 1 2
-_
-5 2 9>3 6>4 8>7 1
-              _
-3 6 8>7>4 9 1<2 5
-EOF
+    expect("futoshiki").to have_output_matching_saved_example
   end
 end

data/spec/integration/knights_puzzle_spec.rb

@@ -1,98 +1,5 @@
-# There are multiple solutions, so this test is nondeterministic
-# This is what Z3 4.8.13 returns
 describe "Knights Swap Puzzle" do
   it do
-    expect("knights_puzzle").to have_output <<EOF
-Solved
-State 0:
-bbb.
-xbxw
-..ww
-x.xw
-w: 2,2 -> 3,0
-
-State 1:
-bbbw
-xbxw
-...w
-x.xw
-b: 1,0 -> 2,2
-
-State 2:
-b.bw
-xbxw
-..bw
-x.xw
-w: 3,1 -> 1,0
-
-State 3:
-bwbw
-xbx.
-..bw
-x.xw
-b: 0,0 -> 1,2
-
-State 4:
-.wbw
-xbx.
-.bbw
-x.xw
-b: 1,2 -> 3,1
-
-State 5:
-.wbw
-xbxb
-..bw
-x.xw
-w: 3,3 -> 1,2
-
-State 6:
-.wbw
-xbxb
-.wbw
-x.x.
-w: 1,2 -> 0,0
-
-State 7:
-wwbw
-xbxb
-..bw
-x.x.
-b: 2,0 -> 1,2
-
-State 8:
-ww.w
-xbxb
-.bbw
-x.x.
-w: 3,2 -> 2,0
-
-State 9:
-wwww
-xbxb
-.bb.
-x.x.
-b: 1,1 -> 3,2
-
-State 10:
-wwww
-x.xb
-.bbb
-x.x.
-b: 1,2 -> 3,3
-
-State 11:
-wwww
-x.xb
-..bb
-x.xb
-w: 3,0 -> 1,1
-
-State 12:
-www.
-xwxb
-..bb
-x.xb
-EOF
+    expect("knights_puzzle").to have_output_matching_saved_example
   end
 end

data/spec/optimize_spec.rb

@@ -2,7 +2,7 @@
 
 # Disabled as it crashes on Z3 4.8.13
 module Z3
-  xdescribe Optimize do
+  describe Optimize do
     let(:optimize) { Optimize.new }
     let(:a) { Z3.Int("a") }
     let(:b) { Z3.Int("b") }
@@ -34,6 +34,15 @@ module Z3
       ])
     end
 
+    it "#assert_soft" do
+      optimize.assert_soft a > 0
+      optimize.assert_soft a < 0
+      optimize.assert_soft a < 10
+      optimize.maximize a
+      expect(optimize).to be_satisfiable
+      expect(optimize.model[a].to_i).to eq 9
+    end
+
     it "#statistics" do
       optimize.assert a + b == 4
       optimize.assert b >= 2
@@ -42,16 +51,6 @@ module Z3
       expect(stats.keys).to match_array(["rlimit count", "max memory", "memory", "num allocs"])
     end
 
-    # This is a very simple example of unknown satisfiablity
-    # so we might need more complex one in the future
-    # Unlike Z3::Solver, this is unknown even 4.6.0
-    it "unknown satisfiability" do
-      optimize.assert a**3 == a
-      expect(optimize.check).to eq(:unknown)
-      expect{optimize.satisfiable?}.to raise_error("Satisfiability unknown")
-      expect{optimize.unsatisfiable?}.to raise_error("Satisfiability unknown")
-    end
-
     it "unknown satisfiability" do
       optimize.assert a**a == a
       expect(optimize.check).to eq(:unknown)

data/spec/set_expr_spec.rb

@@ -17,7 +17,7 @@ module Z3
       )
     end
 
-    if Z3.version >= "4.5"
+    if Z3.version_at_least?(4, 5)
       # Only works in z3 4.5, 4.4 (like on Ubuntu) returns bad stuff
       it "union" do
         expect([

data/spec/solver_spec.rb

@@ -43,7 +43,7 @@ module Z3
     # This is a very simple example of unknown satisfiablity
     # so we might need more complex one in the future
     # This is now satisfiable in 4.6.0
-    if Z3.version >= "4.6"
+    if Z3.version_at_least?(4, 6)
       it "unknown satisfiability (until 4.6 fix)" do
         solver.assert a**3 == a
         expect(solver.check).to eq(:sat)

data/spec/spec_helper.rb

@@ -38,6 +38,20 @@ RSpec::Matchers.define :have_output do |expected|
   end
 end
 
+RSpec::Matchers.define :have_output_matching_saved_example do
+  match do |file_name|
+    executable_path = "#{__dir__}/../examples/#{file_name}"
+    actual = IO.popen("ruby -r./spec/coverage_helper #{executable_path}").read
+    @actual = actual.gsub(/ *$/, "")
+    @expected = Pathname("spec/integration/examples").glob("#{file_name}-*.txt").map(&:read).map{|o| o.gsub(/ *$/, "")}
+    @expected.include?(@actual)
+  end
+
+  failure_message do |actual|
+    "Expected one of saved examples, but got:\n#{@actual}"
+  end
+end
+
 RSpec::Matchers.define :have_output_no_color do |expected|
   match do |file_name|
     executable_path = "#{__dir__}/../examples/#{file_name}"

data/spec/z3_spec.rb

@@ -2,4 +2,14 @@ describe Z3 do
   it "#version return version number of Z3 library" do
     expect(Z3.version).to match(/\A\d+\.\d+\.\d+\.\d+\z/)
   end
+
+  it "#version_at_least return if version matches" do
+    version = Z3.version.split(".").map(&:to_i)
+    expect(Z3.version_at_least?(*version)).to eq(true)
+    expect(Z3.version_at_least?(*version[0,2])).to eq(true)
+    expect(Z3.version_at_least?(*version[0,1])).to eq(true)
+    expect(Z3.version_at_least?(version[0])).to eq(true)
+    expect(Z3.version_at_least?(*version[0,2], 999)).to eq(false)
+    expect(Z3.version_at_least?(999)).to eq(false)
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: z3
 version: !ruby/object:Gem::Version
-  version: 0.0.20220630
+  version: 0.0.20221020
 platform: ruby
 authors:
 - Tomasz Wegrzanowski
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-06-30 00:00:00.000000000 Z
+date: 2022-10-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry
@@ -280,6 +280,14 @@ files:
 - spec/integration/dominion_spec.rb
 - spec/integration/dominosa_spec.rb
 - spec/integration/eulero_spec.rb
+- spec/integration/examples/abc_path-1.txt
+- spec/integration/examples/abc_path-2.txt
+- spec/integration/examples/futoshiki-1.txt
+- spec/integration/examples/futoshiki-2.txt
+- spec/integration/examples/knights_puzzle-1.txt
+- spec/integration/examples/knights_puzzle-2.txt
+- spec/integration/examples/knights_puzzle-3.txt
+- spec/integration/examples/knights_puzzle-4.txt
 - spec/integration/four_hackers_puzzle_spec.rb
 - spec/integration/futoshiki_spec.rb
 - spec/integration/geometry_problem_spec.rb