svgwrite_diagram: add a test function
authorAntonio Ospite <ao2@ao2.it>
Tue, 26 Jun 2018 14:18:31 +0000 (16:18 +0200)
committerAntonio Ospite <ao2@ao2.it>
Wed, 27 Jun 2018 13:18:16 +0000 (15:18 +0200)
Add a test function similar to the one used in the cairo backend.

Add also the same workaround to handle local includes in python2.

src/diagram/svgwrite_diagram.py

index 186d134..bbc88eb 100755 (executable)
@@ -21,7 +21,10 @@ import warnings
 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):
@@ -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
+
+
+def test():
+    diagram = SvgwriteDiagram(400, 400)
+
+    Diagram.test(diagram)
+
+    diagram.save_svg('svgwrite_diagram_test.svg')
+
+
+if __name__ == "__main__":
+    test()