Add basic tests for SVG circles and arrows

fouinou · Jan 1, 2018 · 232a1b1 · 232a1b1
1 parent 6a5210c
commit 232a1b1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test.py b/test.py
@@ -25,6 +25,7 @@
 import chess.syzygy
 import chess.gaviota
 import chess.variant
+import chess.svg
 import collections
 import copy
 import os
@@ -3248,6 +3249,15 @@ def test_svg_board(self):
         self.assertIn("white bishop", svg)
         self.assertNotIn("black queen", svg)
 
+    def test_svg_arrows(self):
+        svg = chess.svg.board(arrows=[(chess.A1, chess.A1)])
+        self.assertIn("<circle", svg)
+        self.assertNotIn("<line", svg)
+
+        svg = chess.svg.board(arrows=[(chess.A1, chess.H8)])
+        self.assertNotIn("<circle", svg)
+        self.assertIn("<line", svg)
+
 
 class SuicideTestCase(unittest.TestCase):