def test_driver_imports_first(self, stdin, stdout):
source = """\
# This will raise a TypeError, unless we patch the sum() function before
# importing this module. example_patching_driver.py does the patch, so
# it has to be imported before this module.
start = sum([1, 2, "3"])
def foo(x):
return x + start
"""
expected_report = """\
start = 99
x = 10
return 109
"""
stdin.read.return_value = source
try:
main()
except SystemExit:
pass
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_canvas_main(self, stdin, stdout):
source = """\
from turtle import *
forward(100)
"""
expected_report = """\
start_canvas
create_line
400
300
500
300
fill='black'
pensize=1
end_canvas
.
"""
stdin.read.return_value = source
main()
report = ''.join(call_args[0][0]
for call_args in stdout.write.call_args_list)
self.assertReportEqual(expected_report, report)
def test_canvas_main(self, stdin, stdout):
source = """\
from turtle import *
forward(100)
"""
expected_report = """\
start_canvas
create_line
400
300
500
300
fill='black'
pensize=1
end_canvas
.
"""
stdin.read.return_value = source
main()
report = ''.join(call_args[0][0]
for call_args in stdout.write.call_args_list)
self.assertReportEqual(expected_report, report)
def test_driver_imports_first(self, stdin, stdout):
source = """\
# This will raise a TypeError, unless we patch the sum() function before
# importing this module. example_patching_driver.py does the patch, so
# it has to be imported before this module.
start = sum([1, 2, "3"])
def foo(x):
return x + start
"""
expected_report = """\
start = 99
x = 10
return 109
"""
stdin.read.return_value = source
try:
main()
except SystemExit:
pass
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_unittest_driver_passes(self, stdin, stdout):
source = """\
from unittest import TestCase
def get_foo(x):
return x + 5
class FooTest(TestCase):
def test_get_foo(self):
y = get_foo(10)
self.assertEqual(15, y)
"""
expected_report = """\
x = 10
return 15
y = 15
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_unittest_driver_passes(self, stdin, stdout):
source = """\
from unittest import TestCase
def get_foo(x):
return x + 5
class FooTest(TestCase):
def test_get_foo(self):
y = get_foo(10)
self.assertEqual(15, y)
"""
expected_report = """\
x = 10
return 15
y = 15
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_main(self, stdin, stdout):
code = """\
i = 1
"""
expected_report = """\
i = 1 """
stdin.read.return_value = code
main()
self.assertEqual([call(expected_report), call("\n")], stdout.write.call_args_list)
def test_dump_arg(self, stdin, stdout):
code = """\
i = 1 + 1
"""
expected_report = """\
i = 1 + 1 | i = 2 """
stdin.read.return_value = code
main()
self.assertEqual([call(expected_report), call('\n')],
stdout.write.call_args_list)
def test_dump_arg(self, stdin, stdout):
code = """\
i = 1 + 1
"""
expected_report = """\
i = 1 + 1 | i = 2 """
stdin.read.return_value = code
main()
self.assertEqual([call(expected_report), call('\n')],
stdout.write.call_args_list)
def test_dunder_file(self, stdin, stdout):
source = """\
filename = __file__
"""
expected_report = """\
filename = '/path/to/foo.py'
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_source_file_arg(self, stdin, stdout):
expected_report = """\
def foo(x): | x = 3
return x + 1 | return 4
|
if __name__ == '__live_coding__': |
y = foo(3) | y = 4
"""
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_dunder_file(self, stdin, stdout):
source = """\
filename = __file__
"""
expected_report = """\
filename = '/path/to/foo.py'
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_syntax_error(self, stdin, stdout):
source = """\
def missing_body():
"""
expected_report = """\
SyntaxError: unexpected EOF while parsing
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_syntax_error(self, stdin, stdout):
source = """\
def missing_body():
"""
expected_report = """\
SyntaxError: unexpected EOF while parsing
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_dunder_file_not_set(self, stdin, stdout):
source = """\
filename = __file__
"""
expected_report = """\
NameError: name '__file__' is not defined
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_dunder_file_not_set(self, stdin, stdout):
source = """\
filename = __file__
"""
expected_report = """\
NameError: name '__file__' is not defined
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_driver_syntax_error(self, stdin, stdout):
source = """\
x = 'Hello, World!'
"""
expected_report = """\
{} line 4: SyntaxError: invalid syntax
""".format(EXAMPLE_DRIVER_SYNTAX_ERROR_PATH)
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_main(self, stdin, stdout):
code = """\
i = 1
name = __name__
"""
expected_report = """\
i = 1
name = '__live_coding__' """
stdin.read.return_value = code
main()
self.assertReportEqual(expected_report,
stdout.write.call_args_list[0][0][0])
def test_driver_syntax_error(self, stdin, stdout):
source = """\
x = 'Hello, World!'
"""
expected_report = """\
{} line 4: SyntaxError: invalid syntax
""".format(EXAMPLE_DRIVER_SYNTAX_ERROR_PATH)
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_main(self, stdin, stdout):
code = """\
i = 1
name = __name__
"""
expected_report = """\
i = 1
name = '__live_coding__' """
stdin.read.return_value = code
main()
self.assertReportEqual(expected_report,
stdout.write.call_args_list[0][0][0])
def test_driver_package(self, stdin, stdout):
source = """\
def foo(x):
return 42
"""
expected_report = """\
x = 'from package __main__.py'
return 42
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_args_no_driver(self, stdin, stdout):
source = """\
import sys
x = sys.argv[1:]
"""
expected_report = """\
x = ['99']
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_driver_package(self, stdin, stdout):
source = """\
def foo(x):
return 42
"""
expected_report = """\
x = 'from package __main__.py'
return 42
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_args_no_driver(self, stdin, stdout):
source = """\
import sys
x = sys.argv[1:]
"""
expected_report = """\
x = ['99']
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_driver_module(self, stdin, stdout):
source = """\
import sys
def foo(x):
return sys.argv[1:]
"""
expected_report = """\
x = 42
return ['99']
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_lib_in_package(self, stdin, stdout):
source = """\
def add_message(s):
package = __package__
return s + ' Received'
"""
expected_report = """\
s = 'from driver'
package = 'example_package'
return 'from driver Received'
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_lib_in_package(self, stdin, stdout):
source = """\
def add_message(s):
package = __package__
return s + ' Received'
"""
expected_report = """\
s = 'from driver'
package = 'example_package'
return 'from driver Received'
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_driver_module(self, stdin, stdout):
source = """\
import sys
def foo(x):
return sys.argv[1:]
"""
expected_report = """\
x = 42
return ['99']
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_driver(self, stdin, stdout):
source = """\
def foo(x):
name = __name__
return x + 1
"""
expected_report = """\
x = 42
name = 'example_source'
return 43
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_driver(self, stdin, stdout):
source = """\
def foo(x):
name = __name__
return x + 1
"""
expected_report = """\
x = 42
name = 'example_source'
return 43
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_silent_driver(self, stdin, stdout):
""" Driver calls code, but doesn't generate messages. """
source = """\
def bar(bucket):
bucket.add('bar')
"""
expected_report = """\
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_unknown_driver(self, stdin, stdout):
source = """\
s = 'Yo!'
"""
expected_report = """\
FileNotFoundError: [Errno 2] No such file or directory: 'bogus_driver.py' |
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_exception_with_driver(self, stdin, stdout):
source = """\
import sys
def foo(x):
sys.exit('Bad stuff.')
"""
expected_report = """\
x = 42
SystemExit: Bad stuff.
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_exception_with_driver(self, stdin, stdout):
source = """\
import sys
def foo(x):
sys.exit('Bad stuff.')
"""
expected_report = """\
---------------------- |
SystemExit: Bad stuff. | x = 42
---------------------- | SystemExit: Bad stuff.
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_unknown_driver(self, stdin, stdout):
source = """\
s = 'Yo!'
"""
expected_report = """\
FileNotFoundError: [Errno 2] No such file or directory: 'bogus_driver.py' |
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_source_file_arg(self, stdin, stdout):
expected_report = """\
def foo(x): | x = 3
return x + 1 | return 4
|
|
def bar(bucket): |
bucket.add('bar') |
|
|
if __name__ == '__live_coding__': |
y = foo(3) | y = 4
"""
stdin.read.return_value = ""
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_doctest_driver_fails(self, stdin, stdout):
source = """\
def get_foo(x):
''' Example for doctest.
>>> get_foo(42)
942
'''
return x + 500
"""
expected_report = """\
------------------------------------------------ |
Traceback (most recent call last): |
File "path/doctest.py", line 9999, in <module> |
sys.exit(_test()) |
SystemExit: 1 |
------------------------------------------------ |
x = 42
return 542
"""
stdin.read.return_value = source
stdout.encoding = None
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
expected_report = self.trim_exception(expected_report)
report = self.trim_exception(report)
self.assertReportEqual(expected_report, report)
def test_bad_driver(self, stdin, stdout):
source = """\
def foo(x):
name = __name__
return x + 1
BAR = 'baz'
"""
expected_report = """\
----------------------------------------------------------------------------------- |
example_driver.py doesn't call the different_source module. Try a different driver. |
----------------------------------------------------------------------------------- |
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_bad_driver_message(self, stdin, stdout):
source = """\
def foo(x):
name = __name__
return x + 1
BAR = 'baz'
"""
expected_report = """\
------------------------------------------------ |
Run config 'example' is bad, try something else. |
------------------------------------------------ |
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_bad_driver(self, stdin, stdout):
source = """\
def foo(x):
name = __name__
return x + 1
BAR = 'baz'
"""
expected_report = """\
----------------------------------------------------------------------------------- |
example_driver.py doesn't call the different_source module. Try a different driver. |
----------------------------------------------------------------------------------- |
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_bad_driver_message(self, stdin, stdout):
source = """\
def foo(x):
name = __name__
return x + 1
BAR = 'baz'
"""
expected_report = """\
------------------------------------------------ |
Run config 'example' is bad, try something else. |
------------------------------------------------ |
"""
stdin.read.return_value = source
main()
report = stdout.write.call_args_list[0][0][0]
self.assertReportEqual(expected_report, report)
def test_driver_fails(self, stdin, stdout):
source = """\
foo = 'Hello, World!'
"""
expected_report = """\
foo = 'Hello, World!' | ---------------------------------------------------- |
| Traceback (most recent call last): |
| File "path/example_driver.py", line 6, in <module> |
| assert 'fail' not in sys.argv, sys.argv[1:] |
| AssertionError: ['fail', 'badly'] |
| ---------------------------------------------------- |
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_driver_fails(self, stdin, stdout):
source = """\
s = 'Hello, World!'
"""
expected_report = """\
s = 'Hello, World!' | ---------------------------------------------------- |
| Traceback (most recent call last): |
| File "path/example_driver.py", line 6, in <module> |
| assert 'fail' not in sys.argv, sys.argv[1:] |
| AssertionError: ['fail', 'badly'] |
| ---------------------------------------------------- |
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_doctest_driver_fails(self, stdin, stdout):
source = """\
def get_foo(x):
''' Example for doctest.
>>> get_foo(42)
942
'''
return x + 500
"""
expected_report = """\
------------------------------------------------ |
SystemExit: 1 |
------------------------------------------------ |
x = 42
return 542
"""
stdin.read.return_value = source
stdout.encoding = None
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
expected_report = self.trim_exception(expected_report)
report = self.trim_exception(report)
self.assertReportEqual(expected_report, report)
def test_doctest_driver_fails(self, stdin, stdout):
source = """\
def get_foo(x):
''' Example for doctest.
>>> get_foo(42)
942
'''
return x + 500
"""
expected_report = """\
------------------------------------------------ |
SystemExit: 1 |
------------------------------------------------ |
x = 42
return 542
"""
stdin.read.return_value = source
stdout.encoding = None
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
expected_report = self.trim_exception(expected_report)
report = self.trim_exception(report)
self.assertReportEqual(expected_report, report)
def test_driver_pycharm_failures(self, stdin, stdout):
""" PyCharm's Pytest wrapper reports failures, but doesn't set exit code.
Look for === FAILURES === report.
"""
source = """\
foo = 'Hello, World!'
"""
expected_report = """\
foo = 'Hello, World!' | ------------------------- |
| Pytest reported failures. |
| ------------------------- |
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_driver_pycharm_failures(self, stdin, stdout):
""" PyCharm's Pytest wrapper reports failures, but doesn't set exit code.
Look for === FAILURES === report.
"""
source = """\
foo = 'Hello, World!'
"""
expected_report = """\
foo = 'Hello, World!' | ------------------------- |
| Pytest reported failures. |
| ------------------------- |
"""
stdin.read.return_value = source
with self.assertRaises(SystemExit):
main()
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_unittest_driver_fails(self, stdin, stdout):
source = """\
from unittest import TestCase
def get_foo(x):
return x + 500
class FooTest(TestCase):
def test_get_foo(self):
y = get_foo(10)
self.assertEqual(15, y)
"""
expected_report = """\
---------------------- |
unittest: (failures=1) |
---------------------- | | x = 10
| return 510
y = 510
AssertionError: 15 != 510
"""
if sys.version_info < (3, 0):
expected_report = expected_report.replace('(failures=1)',
'FAIL ')
stdin.read.return_value = source
with self.assertRaises(SystemExit) as ctx:
main()
self.assertEqual(1, ctx.exception.code)
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)
def test_unittest_driver_fails(self, stdin, stdout):
source = """\
from unittest import TestCase
def get_foo(x):
return x + 500
class FooTest(TestCase):
def test_get_foo(self):
y = get_foo(10)
self.fail(y)
"""
expected_report = """\
---------------- |
SystemExit: True |
---------------- | | x = 10
| return 510
y = 510
AssertionError: 510
"""
if sys.version_info < (3, 0):
expected_report = expected_report.replace('(failures=1)',
'FAIL ')
stdin.read.return_value = source
with self.assertRaises(SystemExit) as ctx:
main()
self.assertEqual(1, ctx.exception.code)
report = stdout.write.call_args_list[0][0][0]
report = self.trim_exception(report)
expected_report = self.trim_exception(expected_report)
self.assertReportEqual(expected_report, report)