118 lines
4.4 KiB
Python
118 lines
4.4 KiB
Python
import unittest
|
|
|
|
from sgf_parsing import (
|
|
parse,
|
|
SgfTree,
|
|
)
|
|
|
|
# Tests adapted from `problem-specifications//canonical-data.json`
|
|
|
|
|
|
class SgfParsingTest(unittest.TestCase):
|
|
def test_empty_input(self):
|
|
input_string = ""
|
|
with self.assertRaises(ValueError) as err:
|
|
parse(input_string)
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "tree missing")
|
|
|
|
def test_tree_with_no_nodes(self):
|
|
input_string = "()"
|
|
with self.assertRaises(ValueError) as err:
|
|
parse(input_string)
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "tree with no nodes")
|
|
|
|
def test_node_without_tree(self):
|
|
input_string = ";"
|
|
with self.assertRaises(ValueError) as err:
|
|
parse(input_string)
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "tree missing")
|
|
|
|
def test_node_without_properties(self):
|
|
input_string = "(;)"
|
|
expected = SgfTree()
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_single_node_tree(self):
|
|
input_string = "(;A[B])"
|
|
expected = SgfTree(properties={"A": ["B"]})
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_multiple_properties(self):
|
|
input_string = "(;A[b]C[d])"
|
|
expected = SgfTree(properties={"A": ["b"], "C": ["d"]})
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_properties_without_delimiter(self):
|
|
input_string = "(;A)"
|
|
with self.assertRaises(ValueError) as err:
|
|
parse(input_string)
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "properties without delimiter")
|
|
|
|
def test_all_lowercase_property(self):
|
|
input_string = "(;a[b])"
|
|
with self.assertRaises(ValueError) as err:
|
|
parse(input_string)
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "property must be in uppercase")
|
|
|
|
def test_upper_and_lowercase_property(self):
|
|
input_string = "(;Aa[b])"
|
|
with self.assertRaises(ValueError) as err:
|
|
parse(input_string)
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "property must be in uppercase")
|
|
|
|
def test_two_nodes(self):
|
|
input_string = "(;A[B];B[C])"
|
|
expected = SgfTree(properties={"A": ["B"]}, children=[SgfTree({"B": ["C"]})])
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_two_child_trees(self):
|
|
input_string = "(;A[B](;B[C])(;C[D]))"
|
|
expected = SgfTree(
|
|
properties={"A": ["B"]},
|
|
children=[SgfTree({"B": ["C"]}), SgfTree({"C": ["D"]})],
|
|
)
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_multiple_property_values(self):
|
|
input_string = "(;A[b][c][d])"
|
|
expected = SgfTree(properties={"A": ["b", "c", "d"]})
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_semicolon_in_property_value_doesn_t_need_to_be_escaped(self):
|
|
input_string = "(;A[a;b][foo]B[bar];C[baz])"
|
|
expected = SgfTree(
|
|
properties={"A": ["a;b", "foo"], "B": ["bar"]},
|
|
children=[SgfTree({"C": ["baz"]})],
|
|
)
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_parentheses_in_property_value_don_t_need_to_be_escaped(self):
|
|
input_string = "(;A[x(y)z][foo]B[bar];C[baz])"
|
|
expected = SgfTree(
|
|
properties={"A": ["x(y)z", "foo"], "B": ["bar"]},
|
|
children=[SgfTree({"C": ["baz"]})],
|
|
)
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_escaped_tab_in_property_value_is_converted_to_space(self):
|
|
input_string = "(;A[hello\\ world])"
|
|
expected = SgfTree(properties={"A": ["hello world"]})
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_escaped_newline_in_property_value_is_converted_to_nothing_at_all(self):
|
|
input_string = "(;A[hello\
|
|
world])"
|
|
expected = SgfTree(properties={"A": ["helloworld"]})
|
|
self.assertEqual(parse(input_string), expected)
|
|
|
|
def test_escaped_t_and_n_in_property_value_are_just_letters_not_whitespace(self):
|
|
input_string = "(;A[\t = t and \n = n])"
|
|
expected = SgfTree(properties={"A": ["t = t and n = n"]})
|
|
self.assertEqual(parse(input_string), expected)
|