##// END OF EJS Templates
cleaner string/bytes solution, closes #4759...
Paul Ivanov -
Show More
@@ -1,54 +1,54 b''
1 1 """Test embedding of IPython"""
2 2
3 3 #-----------------------------------------------------------------------------
4 4 # Copyright (C) 2013 The IPython Development Team
5 5 #
6 6 # Distributed under the terms of the BSD License. The full license is in
7 7 # the file COPYING, distributed as part of this software.
8 8 #-----------------------------------------------------------------------------
9 9
10 10 #-----------------------------------------------------------------------------
11 11 # Imports
12 12 #-----------------------------------------------------------------------------
13 13
14 14 import sys
15 15 import nose.tools as nt
16 16 from IPython.utils.process import process_handler
17 17 from IPython.utils.tempdir import NamedFileInTemporaryDirectory
18 18
19 19 #-----------------------------------------------------------------------------
20 20 # Tests
21 21 #-----------------------------------------------------------------------------
22 22
23 _sample_embed = """
23 _sample_embed = b"""
24 24 from __future__ import print_function
25 25 import IPython
26 26
27 27 a = 3
28 28 b = 14
29 29 print(a, '.', b)
30 30
31 31 IPython.embed()
32 32
33 33 print('bye!')
34 34 """
35 35
36 _exit = "exit\r".encode('UTF-8')
36 _exit = b"exit\r"
37 37
38 38 def test_ipython_embed():
39 39 """test that `IPython.embed()` works"""
40 40 with NamedFileInTemporaryDirectory('file_with_embed.py') as f:
41 f.write(_sample_embed.encode('UTF-8'))
41 f.write(_sample_embed)
42 42 f.flush()
43 43
44 44 # run `python file_with_embed.py`
45 45 cmd = [sys.executable, f.name]
46 46
47 47 _, out, p = process_handler(cmd,
48 48 lambda p: (p.stdin.write(_exit), p.communicate()[:], p))
49 49 std = out[0].decode('UTF-8')
50 50 nt.assert_equal(p.returncode, 0)
51 51 nt.assert_in('3 . 14', std)
52 52 nt.assert_in('IPython', std)
53 53 nt.assert_in('bye!', std)
54 54
General Comments 0
You need to be logged in to leave comments. Login now