Unit test manifold with TF

geomstats · Sep 27, 2018 · 7f09fa4 · 7f09fa4
1 parent 8220025
commit 7f09fa4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_manifold_tensorflow.py b/tests/test_manifold_tensorflow.py
@@ -16,12 +16,11 @@ class TestManifoldMethodsTensorFlow(tf.test.TestCase):
 
     def setUp(self):
         gs.random.seed(1234)
-        self.dimension = gs.random.randint(low=1, high=10)
+        self.dimension = 4
         self.manifold = Manifold(self.dimension)
 
     @classmethod
     def setUpClass(cls):
-        tf.enable_eager_execution()
         os.environ['GEOMSTATS_BACKEND'] = 'tensorflow'
         importlib.reload(gs)
 
@@ -34,7 +33,7 @@ def test_dimension(self):
         result = self.manifold.dimension
         expected = self.dimension
         with self.test_session():
-            self.assertAllClose(gs.eval(result), gs.eval(expected))
+            self.assertAllClose(result, expected)
 
     def test_belongs(self):
         point = gs.array([1, 2, 3])