def test_atleast_2d(self): np.testing.assert_equal(atleast_2d([[1], [2], [3]]), self.res) atleast_2d([[1], [2], [3]])
def test_atleast_1d(self): np.testing.assert_equal(atleast_2d([1, 2, 3]), self.res)
def test_atleast_0d(self): np.testing.assert_equal(atleast_2d(1), np.array([[1]]))