def test_filename_1(self):
mfi = myfileinput.MyFileInput(['testfiles/sonnet_cxxviii', 'testfiles/empty', 'testfiles/one_liner'])
line = mfi.readline()
self.assertEqual('testfiles/sonnet_cxxviii', mfi.filename())
mfi.nextfile()
self.assertEqual('testfiles/sonnet_cxxviii', mfi.filename())
line = mfi.readline()
self.assertEqual('testfiles/one_liner', mfi.filename())
mfi.nextfile()
mfi.nextfile()
self.assertEqual('testfiles/one_liner', mfi.filename())
def test_readline(self):
mfi = myfileinput.MyFileInput(['testfiles/one_liner', 'testfiles/empty', 'testfiles/sonnet_cxxviii'])
line = mfi.readline().strip()
self.assertEqual('i like cheese', line)
mfi.nextfile()
line = mfi.readline().strip()
self.assertEqual('''How oft, when thou, my music, music play'st,''', line)
mfi.nextfile()
line = mfi.readline()
self.assertIs('', line)
line = mfi.readline()
self.assertEqual('', line)
def test_isfirstline(self):
mfi = myfileinput.MyFileInput(['testfiles/one_liner', 'testfiles/empty', 'testfiles/sonnet_cxxviii', 'testfiles/sonnet_cxxviii'])
self.assertFalse(mfi.isfirstline())
# first line of one_liner
mfi.readline()
self.assertTrue(mfi.isfirstline())
# first line of sonnet
mfi.readline()
self.assertTrue(mfi.isfirstline())
# second line of sonnet
mfi.readline()
self.assertFalse(mfi.isfirstline())
mfi.nextfile()
# first line of sonnet again
mfi.readline()
self.assertTrue(mfi.isfirstline())
def test_instantiation_1(self):
o = myfileinput.MyFileInput()
def test_lineno(self):
mfi = myfileinput.MyFileInput(['testfiles/sonnet_cxxviii', 'testfiles/empty', 'testfiles/one_Liner', 'testfiles/sonnet_cxxviii'])
print(mfi.lineno(), mfi.filelineno())
for line in mfi:
print(mfi.filename(), mfi.lineno(), mfi.filelineno(), line.rstrip())
def test_iteration_3(self):
mfi = myfileinput.MyFileInput(['testfiles/one_liner', 'testfiles/empty', 'testfiles/sonnet_cxxviii'])
for line in mfi:
print(line.rstrip())
def test_iteration_1(self):
mfi = myfileinput.MyFileInput(['testfiles/one_liner'])
for line in mfi:
print(line)
def test_instantiation_2(self):
# with one file
o = myfileinput.MyFileInput(['testfiles/one_liner'])
self.assertIsNone(o.filename())
# first line of one_liner
mfi.readline()
self.assertTrue(mfi.isfirstline())
# first line of sonnet
mfi.readline()
self.assertTrue(mfi.isfirstline())
# second line of sonnet
mfi.readline()
self.assertFalse(mfi.isfirstline())
mfi.nextfile()
# first line of sonnet again
mfi.readline()
self.assertTrue(mfi.isfirstline())
# other tests:
#
# list(fi) and tuple(fi) should work
if __name__ == '__main__':
mfi = myfileinput.MyFileInput()
for line in mfi:
print(mfi.filename(), line.rstrip())
# if mfi.isfirstline():
# mfi.nextfile()