* Improve pyGMathView test to compare editor and normal modes

1 files changed, 10 insertions, 12 deletions

diff --git a/test/test_pygmathview.py b/test/test_pygmathview.py
index d94e623..cdcec92 100755
--- a/test/test_pygmathview.py
+++ b/test/test_pygmathview.py
@@ -34,18 +34,16 @@ from pygmathlib_test_formula import make_test_formula
 if __name__ == "__main__":
     window = gtk.Window (gtk.WINDOW_TOPLEVEL)
     window.set_title ("pyGMathView Test")
-
-    mathview = gmathview.MathView ()
-
-    mathview.set_editor_mode (True)
-
-    context = mathview.get_context ()
-
-    formula = make_test_formula (context)
-
-    mathview.set_formula (formula)
-
-    window.add (mathview)
+    vbox = gtk.VBox ()
+    window.add (vbox)
+
+    for editor in [True, False]:
+        mathview = gmathview.MathView ()
+        mathview.set_editor_mode (editor)
+        context = mathview.get_context ()
+        formula = make_test_formula (context)
+        mathview.set_formula (formula)
+        vbox.pack_start (mathview, False, False)
 
     window.connect ("delete-event", gtk.main_quit)