1 files changed, 28 insertions, 0 deletions
diff --git a/src/boost/libs/python/test/numpy/templates.py b/src/boost/libs/python/test/numpy/templates.py
new file mode 100755
index 000000000..8290b13a0
--- /dev/null
+++ b/src/boost/libs/python/test/numpy/templates.py
@@ -0,0 +1,28 @@
+#!/usr/bin/env python
+
+# Copyright Jim Bosch & Ankit Daftery 2010-2012.
+# Distributed under the Boost Software License, Version 1.0.
+#    (See accompanying file LICENSE_1_0.txt or copy at
+#          http://www.boost.org/LICENSE_1_0.txt)
+
+import templates_ext
+import unittest
+import numpy
+
+class TestTemplates(unittest.TestCase):
+
+    def testTemplates(self):
+        for dtype in (numpy.int16, numpy.int32, numpy.float32, numpy.complex128):
+            v = numpy.arange(12, dtype=dtype)
+            for shape in ((12,), (4, 3), (2, 6)):
+                a1 = numpy.zeros(shape, dtype=dtype)
+                a2 = v.reshape(a1.shape)
+                templates_ext.fill(a1)
+                self.assert_((a1 == a2).all())
+        a1 = numpy.zeros((12,), dtype=numpy.float64)
+        self.assertRaises(TypeError, templates_ext.fill, a1)
+        a1 = numpy.zeros((12,2,3), dtype=numpy.float32)
+        self.assertRaises(TypeError, templates_ext.fill, a1)
+
+if __name__=="__main__":
+    unittest.main()