Rename quantization examples

geomstats · Dec 28, 2018 · f57411a · f57411a
1 parent 6ac887b
commit f57411a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/examples/quantization_s1.py → examples/plot_quantization_s1.py b/examples/quantization_s1.py → examples/plot_quantization_s1.py
diff --git a/examples/quantization_s2.py → examples/plot_quantization_s2.py b/examples/quantization_s2.py → examples/plot_quantization_s2.py
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -21,6 +21,8 @@
 import examples.plot_square_h2_poincare_half_plane as plot_square_h2_poincare_half_plane  # NOQA
 import examples.plot_square_h2_klein_disk as plot_square_h2_klein_disk
 import examples.tangent_pca_so3 as tangent_pca_so3
+import examples.plot_quantization_s1 as plot_quantization_s1
+import examples.plot_quantization_s2 as plot_quantization_s2
 import geomstats.tests
 
 
@@ -81,6 +83,14 @@ def test_plot_square_h2_square_klein_disk(self):
     def test_tangent_pca_so3(self):
         tangent_pca_so3.main()
 
+    @geomstats.tests.np_only
+    def test_plot_quantization_s1(self):
+        plot_quantization_s1.main()
+
+    @geomstats.tests.np_only
+    def test_plot_quantization_s2(self):
+        plot_quantization_s2.main()
+
 
 if __name__ == '__main__':
         geomstats.tests.main()