bpython/bpython/test/test_args.py at master · allgo27/bpython · GitHub

History

94 lines (79 loc) · 2.47 KB

Raw

# encoding: utf-8

import re

import subprocess

import sys

import tempfile

from textwrap import dedent

from bpython import args

from bpython.test import FixLanguageTestCase as TestCase, unittest

try:

from nose.plugins.attrib import attr

except ImportError:

def attr(*args, **kwargs):

def identity(func):

return func

return identity

@attr(speed="slow")

class TestExecArgs(unittest.TestCase):

def test_exec_dunder_file(self):

with tempfile.NamedTemporaryFile(mode="w") as f:

f.write(

dedent(

"""\

import sys

sys.stderr.write(__file__)

sys.stderr.flush()"""

)

f.flush()

p = subprocess.Popen(

[sys.executable]

+ ["-m", "bpython.curtsies", f.name],

stderr=subprocess.PIPE,

universal_newlines=True,

)

(_, stderr) = p.communicate()

self.assertEqual(stderr.strip(), f.name)

def test_exec_nonascii_file(self):

with tempfile.NamedTemporaryFile(mode="w") as f:

f.write(

dedent(

"""\

#!/usr/bin/env python

# coding: utf-8

"你好 # nonascii"

"""

)

f.flush()

try:

subprocess.check_call(

[sys.executable, "-m", "bpython.curtsies", f.name]

)

except subprocess.CalledProcessError:

self.fail("Error running module with nonascii characters")

def test_exec_nonascii_file_linenums(self):

with tempfile.NamedTemporaryFile(mode="w") as f:

f.write(

dedent(

"""\

#!/usr/bin/env python

# coding: utf-8

1/0

"""

)

f.flush()

p = subprocess.Popen(

[sys.executable, "-m", "bpython.curtsies", f.name],

stderr=subprocess.PIPE,

universal_newlines=True,

)

(_, stderr) = p.communicate()

self.assertIn("line 3", clean_colors(stderr))

def clean_colors(s):

return re.sub(r"\x1b[^m]*m", "", s)

class TestParse(TestCase):

def test_version(self):

with self.assertRaises(SystemExit):

args.parse(["--version"])

bpython/bpython/test/test_args.py at master · allgo27/bpython · GitHub

Expand file tree

Breadcrumbs

test_args.py

Latest commit

History

Breadcrumbs

test_args.py

File metadata and controls