def test_snippet(self): with mock_stdout() as stdout: entry_point(['trifle', '-i', '(+ 1 2)']) self.assertEqual(stdout.getvalue(), '3\n')
def test_no_args_prints_usage(self): with mock_stdout() as stdout: entry_point([]) self.assertEqual(stdout.getvalue(), USAGE + '\n')