Add missing tell()
method to GeneratorFile and add tests
This commit is contained in:
parent
1e49002458
commit
43e77a7a14
2 changed files with 36 additions and 0 deletions
|
@ -14,37 +14,64 @@ class TestGeneratorFile(unittest.TestCase):
|
|||
|
||||
def test_basic_generator(self):
|
||||
with GeneratorFile(self.sample_generator()) as f:
|
||||
self.assertEquals(0, f.tell())
|
||||
self.assertEquals("thisisatest", f.read())
|
||||
self.assertEquals(len("thisisatest"), f.tell())
|
||||
|
||||
def test_same_lengths(self):
|
||||
with GeneratorFile(self.sample_generator()) as f:
|
||||
self.assertEquals("this", f.read(4))
|
||||
self.assertEquals(4, f.tell())
|
||||
|
||||
self.assertEquals("is", f.read(2))
|
||||
self.assertEquals(6, f.tell())
|
||||
|
||||
self.assertEquals("a", f.read(1))
|
||||
self.assertEquals(7, f.tell())
|
||||
|
||||
self.assertEquals("test", f.read(4))
|
||||
self.assertEquals(11, f.tell())
|
||||
|
||||
def test_indexed_lengths(self):
|
||||
with GeneratorFile(self.sample_generator()) as f:
|
||||
self.assertEquals("thisis", f.read(6))
|
||||
self.assertEquals(6, f.tell())
|
||||
|
||||
self.assertEquals("atest", f.read(5))
|
||||
self.assertEquals(11, f.tell())
|
||||
|
||||
def test_misindexed_lengths(self):
|
||||
with GeneratorFile(self.sample_generator()) as f:
|
||||
self.assertEquals("thisis", f.read(6))
|
||||
self.assertEquals(6, f.tell())
|
||||
|
||||
self.assertEquals("ate", f.read(3))
|
||||
self.assertEquals(9, f.tell())
|
||||
|
||||
self.assertEquals("st", f.read(2))
|
||||
self.assertEquals(11, f.tell())
|
||||
|
||||
self.assertEquals("", f.read(2))
|
||||
self.assertEquals(11, f.tell())
|
||||
|
||||
def test_misindexed_lengths_2(self):
|
||||
with GeneratorFile(self.sample_generator()) as f:
|
||||
self.assertEquals("thisisat", f.read(8))
|
||||
self.assertEquals(8, f.tell())
|
||||
|
||||
self.assertEquals("e", f.read(1))
|
||||
self.assertEquals(9, f.tell())
|
||||
|
||||
self.assertEquals("st", f.read(2))
|
||||
self.assertEquals(11, f.tell())
|
||||
|
||||
self.assertEquals("", f.read(2))
|
||||
self.assertEquals(11, f.tell())
|
||||
|
||||
def test_overly_long(self):
|
||||
with GeneratorFile(self.sample_generator()) as f:
|
||||
self.assertEquals("thisisatest", f.read(60))
|
||||
self.assertEquals(11, f.tell())
|
||||
|
||||
|
||||
class TestUsernameGenerator(unittest.TestCase):
|
||||
|
|
Reference in a new issue