Skip to content
Permalink
Browse files

Debug matrices space + add np_and_pytorch_only

  • Loading branch information...
ninamiolane committed Jun 17, 2019
1 parent 81e01a9 commit afbbf6723f78025879f2d4dc1459f60b60353115
Showing with 9 additions and 0 deletions.
  1. +9 −0 geomstats/tests.py
@@ -46,6 +46,15 @@ def np_and_tf_only(test_item):
return test_item


def np_and_pytorch_only(test_item):
"""Decorator to filter tests for numpy and pytorch only."""
if not (np_backend() or pytorch_backend()):
test_item.__unittest_skip__ = True
test_item.__unittest_skip_why__ = (
'Test for numpy and pytorch backends only.')
return test_item


class DummySession():
def __enter__(self):
pass

0 comments on commit afbbf67

Please sign in to comment.
You can’t perform that action at this time.