Added two more unit tests for the overload decisor.

1 files changed, 14 insertions, 1 deletions

diff --git a/tests/samplebinding/overload_test.py b/tests/samplebinding/overload_test.py
index d3ca66ee3..58a4c99c5 100644
--- a/tests/samplebinding/overload_test.py
+++ b/tests/samplebinding/overload_test.py
@@ -28,7 +28,7 @@
 
 import unittest
 
-from sample import Overload, Point, Size
+from sample import Overload, Point, Polygon, Size
 
 class OverloadTest(unittest.TestCase):
     '''Test case for Overload class'''
@@ -69,6 +69,19 @@ class OverloadTest(unittest.TestCase):
         self.assertEqual(overload.intOverloads(2, 4.5), 3)
         self.assertEqual(overload.intOverloads(Point(0, 0), 3), 1)
 
+    def testIntDoubleOverloads(self):
+        overload = Overload()
+        self.assertEqual(overload.intDoubleOverloads(1, 2), Overload.Function0)
+        self.assertEqual(overload.intDoubleOverloads(1, 2.0), Overload.Function0)
+        self.assertEqual(overload.intDoubleOverloads(1.0, 2), Overload.Function1)
+        self.assertEqual(overload.intDoubleOverloads(1.0, 2.0), Overload.Function1)
+
+    def testWrapperIntIntOverloads(self):
+        overload = Overload()
+        self.assertEqual(overload.wrapperIntIntOverloads(Point(), 1, 2), Overload.Function0)
+        self.assertEqual(overload.wrapperIntIntOverloads(Polygon(), 1, 2), Overload.Function1)
+
+
 if __name__ == '__main__':
     unittest.main()