ceph/src/boost/libs/python/test/numpy/shapes.py

   1 #!/usr/bin/env python
   2
   3 # Copyright Jim Bosch & Ankit Daftery 2010-2012.
   4 # Distributed under the Boost Software License, Version 1.0.
   5 #    (See accompanying file LICENSE_1_0.txt or copy at
   6 #          http://www.boost.org/LICENSE_1_0.txt)
   7
   8 import shapes_ext
   9 import unittest
  10 import numpy
  11
  12 class TestShapes(unittest.TestCase):
  13
  14     def testShapes(self):
  15         a1 = numpy.array([(0,1),(2,3)])
  16         a1_shape = (1,4)
  17         a1 = shapes_ext.reshape(a1,a1_shape)
  18         self.assertEqual(a1_shape,a1.shape)
  19
  20 if __name__=="__main__":
  21     unittest.main()