def test_full_command(self):
setup = ['setup line 1', 'setup line 2']
function = 'function call'
expected = [
'python3', '-m', 'timeit', '-s', setup[0], '-s', setup[1], function
]
self.assertEqual(expected, bench.timeit_command(setup, function))
def test_no_setup(self):
setup = None
function = 'function call'
expected = [
'python3', '-m', 'timeit',
function
]
self.assertEqual(expected, bench.timeit_command(setup, function))
def test_full_command(self):
setup = ['setup line 1', 'setup line 2']
function = 'function call'
expected = [
'python3', '-m', 'timeit',
'-s', setup[0],
'-s', setup[1],
function
]
self.assertEqual(expected, bench.timeit_command(setup, function))
def test_no_function(self):
setup = ['setup line 1', 'setup line 2']
function = None
with self.assertRaises(ValueError):
bench.timeit_command(setup, function)
def test_no_setup(self):
setup = None
function = 'function call'
expected = ['python3', '-m', 'timeit', function]
self.assertEqual(expected, bench.timeit_command(setup, function))
def test_no_function(self):
setup = ['setup line 1', 'setup line 2']
function = None
with self.assertRaises(ValueError):
bench.timeit_command(setup, function)
