Python dsplit Examples

Example #1

File: test_shape_base.py Project: xiaoc10/numpy

 def test_2D_array(self):
     a = np.array([[1, 2, 3, 4], [1, 2, 3, 4]])
     try:
         dsplit(a, 2)
         assert_(0)
     except ValueError:
         pass

Example #2

File: test_shape_base.py Project: SoumitraAgarwal/numpy

 def test_2D_array(self):
     a = np.array([[1, 2, 3, 4], [1, 2, 3, 4]])
     try:
         dsplit(a, 2)
         assert_(0)
     except ValueError:
         pass

Example #3

File: test_shape_base.py Project: FlowkoHinti/Dionysos

 def test_3D_array(self):
     a = np.array([[[1, 2, 3, 4],
                    [1, 2, 3, 4]],
                   [[1, 2, 3, 4],
                    [1, 2, 3, 4]]])
     res = dsplit(a, 2)
     desired = [np.array([[[1, 2], [1, 2]], [[1, 2], [1, 2]]]),
                np.array([[[3, 4], [3, 4]], [[3, 4], [3, 4]]])]
     compare_results(res, desired)

Example #4

File: test_shape_base.py Project: ContinuumIO/numpy

 def test_3D_array(self):
     a = np.array([[[1, 2, 3, 4],
                [1, 2, 3, 4]],
               [[1, 2, 3, 4],
                [1, 2, 3, 4]]])
     res = dsplit(a, 2)
     desired = [np.array([[[1, 2], [1, 2]], [[1, 2], [1, 2]]]),
                np.array([[[3, 4], [3, 4]], [[3, 4], [3, 4]]])]
     compare_results(res, desired)