def test_one(self):
"""First test."""
data = """\
This
is
a multi-line
file"""
h = ParserSupport.UndoHandle(StringIO(data))
self.assertEqual(h.readline(), "This\n")
self.assertEqual(h.peekline(), "is\n")
self.assertEqual(h.readline(), "is\n")
# TODO - Meaning of saveline lacking \n?
h.saveline("saved\n")
self.assertEqual(h.peekline(), "saved\n")
h.saveline("another\n")
self.assertEqual(h.readline(), "another\n")
self.assertEqual(h.readline(), "saved\n")
# Test readlines after saveline
h.saveline("saved again\n")
lines = h.readlines()
self.assertEqual(len(lines), 3)
self.assertEqual(lines[0], "saved again\n")
self.assertEqual(lines[1], "a multi-line\n")
self.assertEqual(lines[2], "file") # no trailing \n
# should be empty now
self.assertEqual(h.readline(), "")
h.saveline("save after empty\n")
self.assertEqual(h.readline(), "save after empty\n")
self.assertEqual(h.readline(), "")
def test_safe_readline(self):
data = """\
This
file"""
h = ParserSupport.UndoHandle(StringIO(data))
safe_readline = ParserSupport.safe_readline
self.assertEqual(safe_readline(h), "This\n")
self.assertEqual(safe_readline(h), "file")
self.assertRaises(ValueError, safe_readline, h)
def test_safe_peekline(self):
safe_peekline = ParserSupport.safe_peekline
data = """\
This
file"""
h = ParserSupport.UndoHandle(StringIO(data))
self.assertEqual(safe_peekline(h), "This\n")
h.readline()
self.assertEqual(safe_peekline(h), "file")
h.readline()
self.assertRaises(ValueError, safe_peekline, h)
h.saveline("hello")
self.assertEqual(safe_peekline(h), "hello")
def test_TaggingConsumer(self):
h = StringIO()
tc = ParserSupport.TaggingConsumer(handle=h, colwidth=5)
tc.start_section()
self.assertEqual(h.getvalue(), "***** start_section\n")
h.seek(0)
h.truncate(0)
tc.test1("myline")
self.assertEqual(h.getvalue(), "test1: myline\n")
h.seek(0)
h.truncate(0)
tc.end_section()
self.assertEqual(h.getvalue(), "***** end_section\n")
def test_attempt_read_and_call(self):
data = """\
>gi|132871|sp|P19947|RL30_BACSU 50S RIBOSOMAL PROTEIN L30 (BL27)
MAKLEITLKRSVIGRPEDQRVTVRTLGLKKTNQTVVHEDNAAIRGMINKVSHLVSVKEQ
>gi|132679|sp|P19946|RL15_BACSU 50S RIBOSOMAL PROTEIN L15
MKLHELKPSEGSRKTRNRVGRGIGSGNGKTAGKGHKGQNARSGGGVRPGFEGGQMPLFQRLPKRGFTNIN
RKEYAVVNLDKLNGFAEGTEVTPELLLETGVISKLNAGVKILGNGKLEKKLTVKANKFSASAKEAVEAAG
GTAEVI"""
h = ParserSupport.UndoHandle(StringIO(data))
lines = []
arac = ParserSupport.attempt_read_and_call
self.assertTrue(arac(h, lines.append, contains="RIBOSOMAL PROTEIN"))
self.assertFalse(arac(h, lines.append, start="foobar"))
self.assertFalse(arac(h, lines.append, blank=1))
self.assertTrue(arac(h, lines.append, end="LVSVKEQ"))
def test_undohandle_read_block(self):
"""Test reading in blocks."""
data = """\
This
is
a multi-line
file"""
for block in [1, 2, 10]:
s = StringIO(data)
h = ParserSupport.UndoHandle(s)
h.peekline()
new = ""
while True:
tmp = h.read(block)
if not tmp:
break
new += tmp
self.assertEqual(data, new)
h.close()
def test_read_and_call(self):
data = """\
>gi|132871|sp|P19947|RL30_BACSU 50S RIBOSOMAL PROTEIN L30 (BL27)
MAKLEITLKRSVIGRPEDQRVTVRTLGLKKTNQTVVHEDNAAIRGMINKVSHLVSVKEQ
>gi|132679|sp|P19946|RL15_BACSU 50S RIBOSOMAL PROTEIN L15
MKLHELKPSEGSRKTRNRVGRGIGSGNGKTAGKGHKGQNARSGGGVRPGFEGGQMPLFQRLPKRGFTNIN
RKEYAVVNLDKLNGFAEGTEVTPELLLETGVISKLNAGVKILGNGKLEKKLTVKANKFSASAKEAVEAAG
GTAEVI
"""
h = ParserSupport.UndoHandle(StringIO(data))
lines = []
rac = ParserSupport.read_and_call
rac(h, lines.append)
self.assertEqual(lines[-1][:10], ">gi|132871")
rac(h, lines.append, start="MAKLE", end="KEQ", contains="SVIG")
self.assertRaises(ValueError, rac, h, lines.append, blank=1)
self.assertRaises(ValueError, rac, h, lines.append, start="foobar")
self.assertRaises(ValueError, rac, h, lines.append, end="foobar")
self.assertRaises(ValueError, rac, h, lines.append, contains="foobar")
self.assertRaises(ValueError, rac, h, lines.append, blank=0)
print(s)
h = TestHandle()
tc = ParserSupport.TaggingConsumer(handle=h, colwidth=5)
tc.start_section() # '***** start_section\n'
tc.test1('myline') # 'test1: myline\n'
tc.end_section() # '***** end_section\n'
# is_blank_line
print("Running tests on is_blank_line")
is_blank_line = lambda *args, **keywds: \
pb(ParserSupport.is_blank_line(*args, **keywds))
print(is_blank_line('\n')) # 1
print(is_blank_line('\r\n')) # 1
print(is_blank_line('\r')) # 1
print(is_blank_line('')) # 1
print(is_blank_line('', allow_spaces=1)) # 1
print(is_blank_line('', allow_spaces=0)) # 1
print(is_blank_line(string.whitespace, allow_spaces=1)) # 1
print(is_blank_line('hello')) # 0
print(is_blank_line('hello', allow_spaces=1)) # 0
print(is_blank_line('hello', allow_spaces=0)) # 0
print(is_blank_line(string.whitespace, allow_spaces=0)) # 0
# safe_readline
def is_blank_line(*args, **keywds):
return pb(ParserSupport.is_blank_line(*args, **keywds))
def arac(*args, **keywds):
return pb(ParserSupport.attempt_read_and_call(*args, **keywds))
return 1
return 0
# TaggingConsumer
print("Running tests on TaggingConsumer")
class TestHandle(object):
def write(self, s):
print(s)
h = TestHandle()
tc = ParserSupport.TaggingConsumer(handle=h, colwidth=5)
tc.start_section() # '***** start_section\n'
tc.test1('myline') # 'test1: myline\n'
tc.end_section() # '***** end_section\n'
# is_blank_line
print("Running tests on is_blank_line")
def is_blank_line(*args, **keywds):
return pb(ParserSupport.is_blank_line(*args, **keywds))
print(is_blank_line('\n')) # 1
def test_read(self):
"""Test read method."""
h = ParserSupport.UndoHandle(StringIO("some text"))
h.saveline("more text")
self.assertEqual(h.read(), "more textsome text")
return 1
return 0
# TaggingConsumer
print("Running tests on TaggingConsumer")
class TestHandle(object):
def write(self, s):
print(s)
h = TestHandle()
tc = ParserSupport.TaggingConsumer(handle=h, colwidth=5)
tc.start_section() # '***** start_section\n'
tc.test1('myline') # 'test1: myline\n'
tc.end_section() # '***** end_section\n'
# is_blank_line
print("Running tests on is_blank_line")
is_blank_line = lambda *args, **keywds: \
pb(ParserSupport.is_blank_line(*args, **keywds))
print(is_blank_line('\n')) # 1
print(is_blank_line('\r\n')) # 1
print(is_blank_line('\r')) # 1
print(is_blank_line('')) # 1
