2
0
mirror of https://github.com/boostorg/python.git synced 2026-01-23 05:42:30 +00:00
Files
python/libs/python/numpy/test/templates.py
2010-06-24 22:20:55 +00:00

18 lines
537 B
Python
Executable File

import templates_mod
import unittest
import numpy
class TestTemplates(unittest.TestCase):
def testTemplates(self):
for dtype in (numpy.int16, numpy.int32, numpy.float32, numpy.complex128):
a1 = numpy.zeros((12,), dtype=dtype)
a2 = numpy.arange(12, dtype=dtype)
templates_mod.fill(a1)
self.assert_((a1 == a2).all())
a1 = numpy.zeros((12,), dtype=numpy.float64)
self.assertRaises(TypeError, templates_mod.fill, a1)
if __name__=="__main__":
unittest.main()