Files
cvc5/examples/api/python/finite_field.py
Alex Ozdemir 368f3c3ed6 ff: connect sub-theories to main solver & test (#9218)
Organizing the PR a bit:

we hook up the subtheories to TheoryFf
we expose FF-related things via the C++/Pytohn API and SMT-LIB2 interface.
we add a bunch of tests against these interfaces.
2022-12-16 16:36:41 +00:00

60 lines
1.5 KiB
Python

###############################################################################
# Top contributors (to current version):
# Andrew Reynolds, Alex Ozdemir
#
# This file is part of the cvc5 project.
#
# Copyright (c) 2009-2022 by the authors listed in the file AUTHORS
# in the top-level source directory and their institutional affiliations.
# All rights reserved. See the file COPYING in the top-level source
# directory for licensing information.
# #############################################################################
#
# A simple test for cvc5's finite field solver.
#
##
import cvc5
from cvc5 import Kind
if __name__ == "__main__":
slv = cvc5.Solver()
slv.setOption("produce-models", "true")
F = slv.mkFiniteFieldSort("5")
a = slv.mkConst(F, "a")
b = slv.mkConst(F, "b")
inv = slv.mkTerm(
Kind.EQUAL,
slv.mkTerm(Kind.FINITE_FIELD_MULT, a, b),
slv.mkFiniteFieldElem("1", F),
)
aIsTwo = slv.mkTerm(
Kind.EQUAL,
a,
slv.mkFiniteFieldElem("2", F),
)
# ab - 1 = 0
slv.assertFormula(inv)
# a = 2
slv.assertFormula(aIsTwo)
r = slv.checkSat()
# should be SAT, with b = 2^(-1)
assert r.isSat()
print(slv.getValue(a).toPythonObj())
assert slv.getValue(b).toPythonObj() == -2
bIsTwo = slv.mkTerm(
Kind.EQUAL,
b,
slv.mkFiniteFieldElem("2", F),
)
# b = 2
slv.assertFormula(bIsTwo)
r = slv.checkSat()
# should be UNSAT, since 2*2 - 1 != 0
assert not r.isSat()