From f04424e0d747d4d6986b9224e3ca8d7e4ad246ac Mon Sep 17 00:00:00 2001
From: ZedongPeng <peng_zedong@126.com>
Date: Tue, 28 Nov 2023 15:57:37 -0500
Subject: [PATCH] add unit test for mindtpy

---
 pyomo/contrib/mindtpy/tests/unit_test.py | 70 ++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 pyomo/contrib/mindtpy/tests/unit_test.py

diff --git a/pyomo/contrib/mindtpy/tests/unit_test.py b/pyomo/contrib/mindtpy/tests/unit_test.py
new file mode 100644
index 00000000000..d9b2e494ab0
--- /dev/null
+++ b/pyomo/contrib/mindtpy/tests/unit_test.py
@@ -0,0 +1,70 @@
+#  ___________________________________________________________________________
+#
+#  Pyomo: Python Optimization Modeling Objects
+#  Copyright (c) 2008-2022
+#  National Technology and Engineering Solutions of Sandia, LLC
+#  Under the terms of Contract DE-NA0003525 with National Technology and
+#  Engineering Solutions of Sandia, LLC, the U.S. Government retains certain
+#  rights in this software.
+#  This software is distributed under the 3-clause BSD License.
+#  ___________________________________________________________________________
+
+import pyomo.common.unittest as unittest
+from pyomo.contrib.mindtpy.util import set_var_value
+
+from pyomo.environ import Var, Integers, ConcreteModel, Integers
+
+
+class UnitTestMindtPy(unittest.TestCase):
+    def test_set_var_value(self):
+        m = ConcreteModel()
+        m.x1 = Var(within=Integers, bounds=(-1, 4), initialize=0)
+
+        set_var_value(
+            m.x1,
+            var_val=5,
+            integer_tolerance=1e-6,
+            zero_tolerance=1e-6,
+            ignore_integrality=False,
+        )
+        self.assertEqual(m.x1.value, 4)
+
+        set_var_value(
+            m.x1,
+            var_val=-2,
+            integer_tolerance=1e-6,
+            zero_tolerance=1e-6,
+            ignore_integrality=False,
+        )
+        self.assertEqual(m.x1.value, -1)
+
+        set_var_value(
+            m.x1,
+            var_val=1.1,
+            integer_tolerance=1e-6,
+            zero_tolerance=1e-6,
+            ignore_integrality=True,
+        )
+        self.assertEqual(m.x1.value, 1.1)
+
+        set_var_value(
+            m.x1,
+            var_val=2.00000001,
+            integer_tolerance=1e-6,
+            zero_tolerance=1e-6,
+            ignore_integrality=False,
+        )
+        self.assertEqual(m.x1.value, 2)
+
+        set_var_value(
+            m.x1,
+            var_val=0.0000001,
+            integer_tolerance=1e-9,
+            zero_tolerance=1e-6,
+            ignore_integrality=False,
+        )
+        self.assertEqual(m.x1.value, 0)
+
+
+if __name__ == '__main__':
+    unittest.main()