def test_string_is_comment_with_two_comment_tokens(self):
"""Test string_is_comment.
When it is a comment with another #
"""
s_input = "# I am a comment #"
self.assertTrue(string_is_comment(s_input))
def test_string_is_comment_when_not_because_ending(self):
"""Test string_is_comment.
When it is not a comment
"""
s_input = "I am not a comment #"
self.assertFalse(string_is_comment(s_input))
def test_string_is_comment(self):
"""Test string_is_comment.
When it is a comment
"""
s_input = "# I am a comment"
self.assertTrue(string_is_comment(s_input))
def process_comments(data, column=None):
"""Reposition comments."""
if isinstance(data, dict):
if data.ca and data.ca.items:
# print("CommentedMap " + str(data.ca.items), file=sys.stderr)
for key in data.ca.items.keys():
if data.ca.items[key][2]:
comment = data.ca.items[key][2].value
fix_for_issue29(data, key)
if string_is_comment(comment):
process_single_comment(data, comment, key, column)
for key, val in data.items():
process_comments(key, column=column)
process_comments(val, column=column)
elif isinstance(data, list):
if data.ca and data.ca.items:
# print("CommentedSeq " + str(data.ca.items), file=sys.stderr)
for key in data.ca.items.keys():
if data.ca.items[key][0]:
comment = data.ca.items[key][0].value
if string_is_comment(comment):
process_single_comment(data, comment, key, column)
for elem in data:
process_comments(elem, column=column)