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 8import shapes_ext 9import unittest 10import numpy 11 12class 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 20if __name__=="__main__": 21 unittest.main() 22