bpython/bpython/test/test_importcompletion.py at master · flyingfishfuse/bpython · GitHub

History

61 lines (48 loc) · 1.62 KB

Raw

# encoding: utf-8

from __future__ import unicode_literals

from bpython import importcompletion

from bpython.test import unittest

class TestSimpleComplete(unittest.TestCase):

def setUp(self):

self.original_modules = importcompletion.modules

importcompletion.modules = [

"zzabc",

"zzabd",

"zzefg",

"zzabc.e",

"zzabc.f",

]

def tearDown(self):

importcompletion.modules = self.original_modules

def test_simple_completion(self):

self.assertSetEqual(

importcompletion.complete(10, "import zza"), set(["zzabc", "zzabd"])

)

def test_package_completion(self):

self.assertSetEqual(

importcompletion.complete(13, "import zzabc."),

set(["zzabc.e", "zzabc.f"]),

)

class TestRealComplete(unittest.TestCase):

@classmethod

def setUpClass(cls):

for _ in importcompletion.find_iterator:

pass

__import__("sys")

__import__("os")

@classmethod

def tearDownClass(cls):

importcompletion.find_iterator = importcompletion.find_all_modules()

importcompletion.modules = set()

def test_from_attribute(self):

self.assertSetEqual(

importcompletion.complete(19, "from sys import arg"), set(["argv"])

)

def test_from_attr_module(self):

self.assertSetEqual(

importcompletion.complete(9, "from os.p"), set(["os.path"])

)

def test_from_package(self):

self.assertSetEqual(

importcompletion.complete(17, "from xml import d"), set(["dom"])

)

bpython/bpython/test/test_importcompletion.py at master · flyingfishfuse/bpython · GitHub

Expand file tree

Breadcrumbs

test_importcompletion.py

Latest commit

History

Breadcrumbs

test_importcompletion.py

File metadata and controls