1 files changed, 27 insertions, 0 deletions
diff --git a/src/cython/test/Mini_simplex_tree_UT.py b/src/cython/test/Mini_simplex_tree_UT.py
new file mode 100755
index 00000000..0d2ce3c3
--- /dev/null
+++ b/src/cython/test/Mini_simplex_tree_UT.py
@@ -0,0 +1,27 @@
+import unittest
+
+import gudhi
+
+class TestMiniSimplexTree(unittest.TestCase):
+
+  def test_mini(self):
+    triangle012 = [0,1,2]
+    edge03 = [0,3]
+    mini_st = gudhi.MiniSimplexTree()
+    self.assertTrue(mini_st.insert(triangle012))
+    self.assertTrue(mini_st.insert(edge03))
+    # FIXME: Remove this line
+    mini_st.set_dimension(2);
+
+    edge02 = [0,2]
+    self.assertTrue(mini_st.find(edge02))
+    self.assertEqual(mini_st.get_coface_tree(edge02, 1), [([0, 1, 2], 0.0)])
+
+    # remove_maximal_simplex test
+    self.assertEqual(mini_st.get_coface_tree(triangle012, 1), [])
+    mini_st.remove_maximal_simplex(triangle012)
+    self.assertTrue(mini_st.find(edge02))
+    self.assertFalse(mini_st.find(triangle012))
+
+if __name__ == '__main__':
+    unittest.main()
+\ No newline at end of file