def test_main_skipheader(self):
"""
test_main_skipheader
"""
self.sin.write('pid\n5\n3')
self.sin.seek(0)
cli.main(['-hl', 'row[0]*2'])
self.assertWasStreamed('10\n6\n')
def test_in_place_no_extension(self):
t = NamedTemporaryFile()
t.write('bye\n')
t.flush()
extension = ''
cli.main(['-l', '-i', extension, 'line.replace("bye", "hi")', t.name])
with open(t.name) as f:
self.assertEqual(f.read(), 'hi\n')
def test_main_linemode(self):
"""
test_main_linemode
"""
self.sin.write('file1\nfile2\nfile3')
self.sin.seek(0)
cli.main(['-l', 'line+".txt"'])
self.assertWasStreamed('file1.txt\nfile2.txt\nfile3.txt\n')
def test_main_initial(self):
"""
test_main_initial
"""
self.assertRaises(NameError, cli.main, ['fake_object'])
cli.main(['-b', 'fake_object=5', 'fake_object*2'])
self.assertWasStreamed('10\n')
def test_passed_file_twice(self):
"""
test_passed_file_twice
"""
t = NamedTemporaryFile()
t.write(b'bye\n')
t.flush()
cli.main(['-l', 'line.replace("bye", "hi")', t.name, t.name])
self.assertWasStreamed('hi\nhi\n')
def test_in_place_modification(self):
t = NamedTemporaryFile()
t.write('bye\n')
t.flush()
extension = '.bak'
backup = t.name + extension
cli.main(['-l', '-i', extension, 'line.replace("bye", "hi")', t.name])
with open(backup) as f:
self.assertEqual(f.read(), 'bye\n')
with open(t.name) as f:
self.assertEqual(f.read(), 'hi\n')
def test_in_place_no_extension(self):
"""
test_in_place_no_extension
"""
t = NamedTemporaryFile()
t.write(b'bye\n')
t.flush()
extension = ''
cli.main(['-l', '-i', extension, 'line.replace("bye", "hi")', t.name])
with open(t.name) as f:
self.assertEqual(f.read(), 'hi\n')
def test_main(self):
cli.main(['range(3)'])
self.assertWasStreamed('0\n1\n2\n')
def test_passed_file_twice(self):
t = NamedTemporaryFile()
t.write('bye\n')
t.flush()
cli.main(['-l', 'line.replace("bye", "hi")', t.name, t.name])
self.assertWasStreamed('hi\nhi\n')
def test_version(self):
cli.main(['--version'])
self.sout.seek(0)
streamed = self.sout.read()
self.assertTrue(re.match('\d+\.\d+\.\d+$', streamed))
def test_main_skipheader(self):
self.sin.write('pid\n5\n3')
self.sin.seek(0)
cli.main(['-hl', 'row[0]*2'])
self.assertWasStreamed('10\n6\n')
def test_main_linemode(self):
self.sin.write('file1\nfile2\nfile3')
self.sin.seek(0)
cli.main(['-l', 'line+".txt"'])
self.assertWasStreamed('file1.txt\nfile2.txt\nfile3.txt\n')
def test_main_raw(self):
cli.main(['-r', 'range(3)'])
self.assertWasStreamed('[0, 1, 2]\n')
def test_main_initial(self):
self.assertRaises(NameError, cli.main, ['fake_object'])
cli.main(['-b', 'fake_object=5', 'fake_object*2'])
self.assertWasStreamed('10\n')
def test_main_raw(self):
"""
test_main_raw
"""
cli.main(['-r', 'list(range(3))'])
self.assertWasStreamed('[0, 1, 2]\n')
def test_main(self):
"""
test_main
"""
cli.main(['range(3)'])
self.assertWasStreamed('0\n1\n2\n')