svgwrite_diagram: add a test function

[flexagon-toolkit.git] / src / diagram / svgwrite_diagram.py

diff --git a/src/diagram/svgwrite_diagram.py b/src/diagram/svgwrite_diagram.py
index 186d134..bbc88eb 100755 (executable)
--- a/src/diagram/svgwrite_diagram.py
+++ b/src/diagram/svgwrite_diagram.py
@@ -21,7 +21,10 @@ import warnings
 from math import degrees
 import svgwrite
 from svgwrite.data.types import SVGAttribute
 from math import degrees
 import svgwrite
 from svgwrite.data.types import SVGAttribute

-from .diagram import Diagram
+try:
+    from .diagram import Diagram
+except ValueError:
+    from diagram import Diagram

 
 
 class InkscapeDrawing(svgwrite.Drawing):
 
 
 class InkscapeDrawing(svgwrite.Drawing):


@@ -189,3 +192,15 @@ class SvgwriteDiagram(Diagram):
 
         # XXX: this is just a very quick and dirty estimate to advance the text
         return self.font_size * len(text) / 2
 
         # XXX: this is just a very quick and dirty estimate to advance the text
         return self.font_size * len(text) / 2

+
+
+def test():
+    diagram = SvgwriteDiagram(400, 400)
+
+    Diagram.test(diagram)
+
+    diagram.save_svg('svgwrite_diagram_test.svg')
+
+
+if __name__ == "__main__":
+    test()