def test_3D_array(self):
     a = array([[1, 2], [1, 2]])
     b = array([[2, 3], [2, 3]])
     a = array([a, a])
     b = array([b, b])
     res = [atleast_2d(a), atleast_2d(b)]
     desired = [a, b]
     assert_array_equal(res, desired)

 def test_r2array(self):
     """ Test to make sure equivalent Travis O's r2array function
     """
     assert_(atleast_2d(3).shape == (1,1))
     assert_(atleast_2d([3j,1]).shape == (1,2))
     assert_(atleast_2d([[[3,1],[4,5]],[[3,5],[1,2]]]).shape == (2,2,2))

 def test_1D_array(self):
     a = array([1, 2])
     b = array([2, 3])
     res = [atleast_2d(a), atleast_2d(b)]
     desired = [array([[1, 2]]), array([[2, 3]])]
     assert_array_equal(res, desired)

 def test_0D_array(self):
     a = array(1)
     b = array(2)
     res = [atleast_2d(a), atleast_2d(b)]
     desired = [array([[1]]), array([[2]])]
     assert_array_equal(res, desired)