Skip to content

Commit

Permalink
Fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
gwtnb committed Aug 10, 2017
1 parent 431afae commit f4f34b8
Showing 1 changed file with 11 additions and 11 deletions.
22 changes: 11 additions & 11 deletions tests/cupy_tests/manipulation_tests/test_dims.py
Expand Up @@ -152,58 +152,58 @@ def test_squeeze2(self, xp):
return a.squeeze()

@testing.numpy_cupy_array_equal()
def test_squeze_int_axis1(self, xp):
def test_squeeze_int_axis1(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
return a.squeeze(axis=2)

@testing.numpy_cupy_array_equal()
def test_squeze_int_axis2(self, xp):
def test_squeeze_int_axis2(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
return a.squeeze(axis=-3)

@testing.with_requires('numpy>=1.13')
@testing.numpy_cupy_raises()
def test_squeze_int_axis_failure1(self, xp):
def test_squeeze_int_axis_failure1(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
a.squeeze(axis=-9)

def test_squeze_int_axis_failure2(self):
def test_squeeze_int_axis_failure2(self):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), cupy)
with self.assertRaises(cupy.core.core._AxisError):
a.squeeze(axis=-9)

@testing.numpy_cupy_array_equal()
def test_squeze_tuple_axis1(self, xp):
def test_squeeze_tuple_axis1(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
return a.squeeze(axis=(2, 4))

@testing.numpy_cupy_array_equal()
def test_squeze_tuple_axis2(self, xp):
def test_squeeze_tuple_axis2(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
return a.squeeze(axis=(-4, -3))

@testing.numpy_cupy_array_equal()
def test_squeze_tuple_axis3(self, xp):
def test_squeeze_tuple_axis3(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
return a.squeeze(axis=(4, 2))

@testing.numpy_cupy_array_equal()
def test_squeze_tuple_axis4(self, xp):
def test_squeeze_tuple_axis4(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
return a.squeeze(axis=())

@testing.with_requires('numpy>=1.13')
@testing.numpy_cupy_raises()
def test_squeze_tuple_axis_failure1(self, xp):
def test_squeeze_tuple_axis_failure1(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
a.squeeze(axis=(-9,))

@testing.numpy_cupy_raises()
def test_squeze_tuple_axis_failure2(self, xp):
def test_squeeze_tuple_axis_failure2(self, xp):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), xp)
a.squeeze(axis=(2, 2))

def test_squeze_tuple_axis_failure3(self):
def test_squeeze_tuple_axis_failure3(self):
a = testing.shaped_arange((1, 2, 1, 3, 1, 1, 4, 1), cupy)
with self.assertRaises(cupy.core.core._AxisError):
a.squeeze(axis=(-9,))
Expand Down

0 comments on commit f4f34b8

Please sign in to comment.