aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/tools/cython/Cython/Tests/TestStringIOTree.py
diff options
context:
space:
mode:
authorAnton Samokhvalov <pg83@yandex.ru>2022-02-10 16:45:15 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:45:15 +0300
commit72cb13b4aff9bc9cf22e49251bc8fd143f82538f (patch)
treeda2c34829458c7d4e74bdfbdf85dff449e9e7fb8 /contrib/tools/cython/Cython/Tests/TestStringIOTree.py
parent778e51ba091dc39e7b7fcab2b9cf4dbedfb6f2b5 (diff)
downloadydb-72cb13b4aff9bc9cf22e49251bc8fd143f82538f.tar.gz
Restoring authorship annotation for Anton Samokhvalov <pg83@yandex.ru>. Commit 1 of 2.
Diffstat (limited to 'contrib/tools/cython/Cython/Tests/TestStringIOTree.py')
-rw-r--r--contrib/tools/cython/Cython/Tests/TestStringIOTree.py132
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 a15f2cd88d..724920085e 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')