diff options
author | Anton Samokhvalov <[email protected]> | 2022-02-10 16:45:17 +0300 |
---|---|---|
committer | Daniil Cherednik <[email protected]> | 2022-02-10 16:45:17 +0300 |
commit | d3a398281c6fd1d3672036cb2d63f842d2cb28c5 (patch) | |
tree | dd4bd3ca0f36b817e96812825ffaf10d645803f2 /contrib/tools/cython/Cython/Tests/TestStringIOTree.py | |
parent | 72cb13b4aff9bc9cf22e49251bc8fd143f82538f (diff) |
Restoring authorship annotation for Anton Samokhvalov <[email protected]>. Commit 2 of 2.
Diffstat (limited to 'contrib/tools/cython/Cython/Tests/TestStringIOTree.py')
-rw-r--r-- | contrib/tools/cython/Cython/Tests/TestStringIOTree.py | 132 |
1 files changed, 66 insertions, 66 deletions
diff --git a/contrib/tools/cython/Cython/Tests/TestStringIOTree.py b/contrib/tools/cython/Cython/Tests/TestStringIOTree.py index 724920085ee..a15f2cd88d7 100644 --- a/contrib/tools/cython/Cython/Tests/TestStringIOTree.py +++ b/contrib/tools/cython/Cython/Tests/TestStringIOTree.py @@ -1,67 +1,67 @@ -import unittest - -from Cython import StringIOTree as stringtree - -code = """ -cdef int spam # line 1 - -cdef ham(): - a = 1 - b = 2 - c = 3 - d = 4 - -def eggs(): - pass - -cpdef bacon(): - print spam - print 'scotch' - print 'tea?' - print 'or coffee?' # line 16 -""" - -linemap = dict(enumerate(code.splitlines())) - -class TestStringIOTree(unittest.TestCase): - - def setUp(self): - self.tree = stringtree.StringIOTree() - - def test_markers(self): - assert not self.tree.allmarkers() - - def test_insertion(self): - self.write_lines((1, 2, 3)) - line_4_to_6_insertion_point = self.tree.insertion_point() - self.write_lines((7, 8)) - line_9_to_13_insertion_point = self.tree.insertion_point() - self.write_lines((14, 15, 16)) - - line_4_insertion_point = line_4_to_6_insertion_point.insertion_point() - self.write_lines((5, 6), tree=line_4_to_6_insertion_point) - - line_9_to_12_insertion_point = ( - line_9_to_13_insertion_point.insertion_point()) - self.write_line(13, tree=line_9_to_13_insertion_point) - - self.write_line(4, tree=line_4_insertion_point) - self.write_line(9, tree=line_9_to_12_insertion_point) - line_10_insertion_point = line_9_to_12_insertion_point.insertion_point() - self.write_line(11, tree=line_9_to_12_insertion_point) - self.write_line(10, tree=line_10_insertion_point) - self.write_line(12, tree=line_9_to_12_insertion_point) - +import unittest + +from Cython import StringIOTree as stringtree + +code = """ +cdef int spam # line 1 + +cdef ham(): + a = 1 + b = 2 + c = 3 + d = 4 + +def eggs(): + pass + +cpdef bacon(): + print spam + print 'scotch' + print 'tea?' + print 'or coffee?' # line 16 +""" + +linemap = dict(enumerate(code.splitlines())) + +class TestStringIOTree(unittest.TestCase): + + def setUp(self): + self.tree = stringtree.StringIOTree() + + def test_markers(self): + assert not self.tree.allmarkers() + + def test_insertion(self): + self.write_lines((1, 2, 3)) + line_4_to_6_insertion_point = self.tree.insertion_point() + self.write_lines((7, 8)) + line_9_to_13_insertion_point = self.tree.insertion_point() + self.write_lines((14, 15, 16)) + + line_4_insertion_point = line_4_to_6_insertion_point.insertion_point() + self.write_lines((5, 6), tree=line_4_to_6_insertion_point) + + line_9_to_12_insertion_point = ( + line_9_to_13_insertion_point.insertion_point()) + self.write_line(13, tree=line_9_to_13_insertion_point) + + self.write_line(4, tree=line_4_insertion_point) + self.write_line(9, tree=line_9_to_12_insertion_point) + line_10_insertion_point = line_9_to_12_insertion_point.insertion_point() + self.write_line(11, tree=line_9_to_12_insertion_point) + self.write_line(10, tree=line_10_insertion_point) + self.write_line(12, tree=line_9_to_12_insertion_point) + self.assertEqual(self.tree.allmarkers(), list(range(1, 17))) - self.assertEqual(code.strip(), self.tree.getvalue().strip()) - - - def write_lines(self, linenos, tree=None): - for lineno in linenos: - self.write_line(lineno, tree=tree) - - def write_line(self, lineno, tree=None): - if tree is None: - tree = self.tree - tree.markers.append(lineno) - tree.write(linemap[lineno] + '\n') + self.assertEqual(code.strip(), self.tree.getvalue().strip()) + + + def write_lines(self, linenos, tree=None): + for lineno in linenos: + self.write_line(lineno, tree=tree) + + def write_line(self, lineno, tree=None): + if tree is None: + tree = self.tree + tree.markers.append(lineno) + tree.write(linemap[lineno] + '\n') |