added a test case for Quantity_Color/Quantity_NameOfColor offset issue

tpaviot · tpaviot · commit 12c1deb67fcf · 2020-11-11T13:02:49.000+01:00
diff --git a/test/core_wrapper_features_unittest.py b/test/core_wrapper_features_unittest.py
@@ -74,6 +74,7 @@
                                TopTools_HSequenceOfShape,
                                TopTools_ListOfShape)
 from OCC.Core.TDF import TDF_LabelNode
+from OCC.Core.Quantity import Quantity_NameOfColor, Quantity_Color
 from OCC.Core.Exception import (methodnotwrapped, classnotwrapped,
                                 MethodNotWrappedError, ClassNotWrappedError)
 
@@ -840,6 +841,26 @@ def test_BRepClass_FaceClassifier(self) -> None:
         bfc = BRepClass_FaceClassifier()
         self.assertIsInstance(bfc, BRepClass_FaceClassifier)
 
+    def test_wrap_enum(self) -> None:
+        """ check that enum integer values match c++ name
+        """
+        red = Quantity_Color(Quantity_NameOfColor.Quantity_NOC_RED)
+        self.assertEqual(red.Red(), 1.)
+        self.assertEqual(red.Green(), 0.)
+        self.assertEqual(red.Blue(), 0.)
+        green = Quantity_Color(Quantity_NameOfColor.Quantity_NOC_GREEN)
+        self.assertEqual(green.Red(), 0.)
+        self.assertEqual(green.Green(), 1.)
+        self.assertEqual(green.Blue(), 0.)
+        blue = Quantity_Color(Quantity_NameOfColor.Quantity_NOC_BLUE)
+        self.assertEqual(blue.Red(), 0.)
+        self.assertEqual(blue.Green(), 0.)
+        self.assertEqual(blue.Blue(), 1.)
+        white = Quantity_Color(Quantity_NameOfColor.Quantity_NOC_WHITE)
+        self.assertEqual(white.Red(), 1.)
+        self.assertEqual(white.Green(), 1.)
+        self.assertEqual(white.Blue(), 1.)
+
 def suite() -> unittest.TestSuite:
     test_suite = unittest.TestSuite()
     test_suite.addTest(unittest.makeSuite(TestWrapperFeatures))
