1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
# Copyright 1999 by Jeffrey Chang. All rights reserved.
# This code is part of the Biopython distribution and governed by its
# license. Please see the LICENSE file that should have been included
# as part of this package.
from __future__ import print_function
import os.path
import unittest
import shutil
from Bio._py3k import StringIO
import tempfile
from Bio import File
data = """This
is
a multi-line
file"""
class UndoHandleTests(unittest.TestCase):
def test_one(self):
h = File.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_read(self):
"""test read method"""
h = File.UndoHandle(StringIO("some text"))
h.saveline("more text")
self.assertEqual(h.read(), 'more textsome text')
def test_undohandle_read_block(self):
for block in [1, 2, 10]:
s = StringIO(data)
h = File.UndoHandle(s)
h.peekline()
new = ""
while True:
tmp = h.read(block)
if not tmp:
break
new += tmp
self.assertEqual(data, new)
h.close()
class AsHandleTestCase(unittest.TestCase):
def setUp(self):
# Create a directory to work in
self.temp_dir = tempfile.mkdtemp(prefix='biopython-test')
def tearDown(self):
shutil.rmtree(self.temp_dir)
def _path(self, *args):
return os.path.join(self.temp_dir, *args)
def test_handle(self):
"Test as_handle with a file-like object argument"
p = self._path('test_file.fasta')
with open(p, 'wb') as fp:
with File.as_handle(fp) as handle:
self.assertEqual(fp, handle, "as_handle should "
"return argument when given a file-like object")
self.assertFalse(handle.closed)
self.assertFalse(handle.closed,
"Exiting as_handle given a file-like object should not "
"close the file")
def test_path(self):
"Test as_handle with a path argument"
p = self._path('test_file.fasta')
mode = 'wb'
with File.as_handle(p, mode=mode) as handle:
self.assertEqual(p, handle.name)
self.assertEqual(mode, handle.mode)
self.assertFalse(handle.closed)
self.assertTrue(handle.closed)
def test_stringio(self):
s = StringIO()
with File.as_handle(s) as handle:
self.assertEqual(s, handle)
if __name__ == "__main__":
runner = unittest.TextTestRunner(verbosity=2)
unittest.main(testRunner=runner)
|
