Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 5f417b5d authored by Soumith Chintala's avatar Soumith Chintala
Browse files

add some scipy skip annotations to tests

parent 7aef257b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
test/test_transforms.py
View file @ 5f417b5d
...@@ -335,6 +335,7 @@ class Tester(unittest.TestCase): ...@@ -335,6 +335,7 @@ class Tester(unittest.TestCase):
# Checking if Lambda can be printed as string # Checking if Lambda can be printed as string
trans.__repr__() trans.__repr__()
@unittest.skipIf(stats is None, 'scipy.stats not available')
def test_random_apply(self): def test_random_apply(self):
random_state = random.getstate() random_state = random.getstate()
random.seed(42) random.seed(42)
...@@ -360,6 +361,7 @@ class Tester(unittest.TestCase): ...@@ -360,6 +361,7 @@ class Tester(unittest.TestCase):
# Checking if RandomApply can be printed as string # Checking if RandomApply can be printed as string
random_apply_transform.__repr__() random_apply_transform.__repr__()
@unittest.skipIf(stats is None, 'scipy.stats not available')
def test_random_choice(self): def test_random_choice(self):
random_state = random.getstate() random_state = random.getstate()
random.seed(42) random.seed(42)
...@@ -395,6 +397,7 @@ class Tester(unittest.TestCase): ...@@ -395,6 +397,7 @@ class Tester(unittest.TestCase):
# Checking if RandomChoice can be printed as string # Checking if RandomChoice can be printed as string
random_choice_transform.__repr__() random_choice_transform.__repr__()
@unittest.skipIf(stats is None, 'scipy.stats not available')
def test_random_order(self): def test_random_order(self):
random_state = random.getstate() random_state = random.getstate()
random.seed(42) random.seed(42)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment