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()