Skip to content

GitLab

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