aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorInigoGutierrez <inigogf.95@gmail.com>2021-01-24 20:47:26 +0100
committerInigoGutierrez <inigogf.95@gmail.com>2021-01-24 20:59:17 +0100
commitddde2a9a43daf870c26bef33f47abe45b414c3d0 (patch)
tree52576f8a97011f733360bc9904f1890867cd8bfd /tests
downloadimago-ddde2a9a43daf870c26bef33f47abe45b414c3d0.tar.gz
imago-ddde2a9a43daf870c26bef33f47abe45b414c3d0.zip
First commit!
Diffstat (limited to 'tests')
-rw-r--r--tests/test_parseHelpers.py99
1 files changed, 99 insertions, 0 deletions
diff --git a/tests/test_parseHelpers.py b/tests/test_parseHelpers.py
new file mode 100644
index 0000000..cf1fa9f
--- /dev/null
+++ b/tests/test_parseHelpers.py
@@ -0,0 +1,99 @@
+"""Tests for parseHelpers module."""
+
+import unittest
+
+from imago.data.enums import Player
+from imago.engine import parseHelpers
+
+TEST_BOARD_SIZE = 19
+
+class TestParseHelpers(unittest.TestCase):
+ """Test parseHelpers module."""
+
+ def testParseMove(self):
+ """Test parsing of GtpMove"""
+
+ self.assertEqual(parseHelpers.parseMove(["B"], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.parseMove(["A1"], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.parseMove(["B", "A1", "W"], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+
+ parsedMove = parseHelpers.parseMove(["B", "t1"], TEST_BOARD_SIZE)
+
+ targetMove = parseHelpers.GtpMove(Player.BLACK, [18, 18])
+ self.assertEqual(parsedMove.color, targetMove.color)
+ self.assertEqual(parsedMove.vertex, targetMove.vertex)
+
+ def testParseColor(self):
+ """Test parsing of color"""
+ self.assertEqual(parseHelpers.parseColor("b"), Player.BLACK)
+ self.assertEqual(parseHelpers.parseColor("B"), Player.BLACK)
+ self.assertEqual(parseHelpers.parseColor("w"), Player.WHITE)
+ self.assertEqual(parseHelpers.parseColor("W"), Player.WHITE)
+ self.assertEqual(parseHelpers.parseColor("bw"), parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.parseColor("wb"), parseHelpers.ParseCodes.ERROR)
+
+ def testParseVertexWrongInputs(self):
+ """Test wrong inputs for parseVertex."""
+ inputs = ( "a", "1", "1a", "aa1", "a0", "u1", "a"+str(TEST_BOARD_SIZE+1) )
+ for text in inputs:
+ self.assertEqual(parseHelpers.parseVertex(text, TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+
+ def testParseVertexCorrectInputs(self):
+ """Test correct inputs and their resulting coordinates for parseVertex."""
+ self.assertEqual(parseHelpers.parseVertex(
+ "a1", TEST_BOARD_SIZE),
+ [18,0])
+ self.assertEqual(parseHelpers.parseVertex(
+ "b1", TEST_BOARD_SIZE),
+ [18,1])
+ self.assertEqual(parseHelpers.parseVertex(
+ "a2", TEST_BOARD_SIZE),
+ [17,0])
+ self.assertEqual(parseHelpers.parseVertex(
+ "b2", TEST_BOARD_SIZE),
+ [17,1])
+
+ self.assertEqual(parseHelpers.parseVertex(
+ "T1", TEST_BOARD_SIZE),
+ [18,18])
+ self.assertEqual(parseHelpers.parseVertex(
+ "T19", TEST_BOARD_SIZE),
+ [0,18])
+ self.assertEqual(parseHelpers.parseVertex(
+ "A19", TEST_BOARD_SIZE),
+ [0,0])
+
+ def testVertexToString(self):
+ """Test converting vertices to strings."""
+ self.assertEqual(parseHelpers.vertexToString([0,0], TEST_BOARD_SIZE), "A19")
+ self.assertEqual(parseHelpers.vertexToString([1,0], TEST_BOARD_SIZE), "A18")
+ self.assertEqual(parseHelpers.vertexToString([2,0], TEST_BOARD_SIZE), "A17")
+ self.assertEqual(parseHelpers.vertexToString([0,1], TEST_BOARD_SIZE), "B19")
+ self.assertEqual(parseHelpers.vertexToString([0,2], TEST_BOARD_SIZE), "C19")
+ self.assertEqual(parseHelpers.vertexToString([0,18], TEST_BOARD_SIZE), "T19")
+ self.assertEqual(parseHelpers.vertexToString([18,0], TEST_BOARD_SIZE), "A1")
+ self.assertEqual(parseHelpers.vertexToString([18,18], TEST_BOARD_SIZE), "T1")
+
+ self.assertEqual(parseHelpers.vertexToString([-1,0], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.vertexToString([0,-1], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.vertexToString([-1,-1], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.vertexToString([19,0], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.vertexToString([0,19], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.vertexToString([19,19], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.vertexToString([0], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+ self.assertEqual(parseHelpers.vertexToString([0,0,0], TEST_BOARD_SIZE),
+ parseHelpers.ParseCodes.ERROR)
+
+if __name__ == '__main__':
+ unittest.main()