60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""
|
||
|
|
test_py-polymorphic-loader
|
||
|
|
----------------------------------
|
||
|
|
|
||
|
|
Tests for `py-polymorphic-loader` module.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
from py_polymorphic_loader import polymorph
|
||
|
|
|
||
|
|
# Note that these compilers don't actually load anything out of the
|
||
|
|
# source files. That's not the point. The point is to show that the
|
||
|
|
# correct compiler has been found for a given extension.
|
||
|
|
|
||
|
|
def compiler2(filename):
|
||
|
|
return compile("result='Success for %s'" % (filename), filename, "exec")
|
||
|
|
|
||
|
|
def compiler3(filename):
|
||
|
|
return compile("result='Success for %s'" % (filename), filename, "exec")
|
||
|
|
|
||
|
|
class Test_Polymorph_1(object):
|
||
|
|
def test_import1(self):
|
||
|
|
import polytestmix
|
||
|
|
polytestmix.install(compiler2, ['.2'])
|
||
|
|
polytestmix.install(compiler3, ['.3'])
|
||
|
|
assert(polytestmix.test2.result = "Success for test2")
|
||
|
|
assert(polytestmix.test3.result = "Success for test3")
|
||
|
|
|
||
|
|
class Test_Polymorph_2(object):
|
||
|
|
def test_import2(self):
|
||
|
|
import polytestmix
|
||
|
|
polytestmix.install(compiler2, ['.2'])
|
||
|
|
polytestmix.install(compiler3, ['.3'])
|
||
|
|
assert(polytestmix.test2.result = "Success for test2")
|
||
|
|
|
||
|
|
class Test_Polymorph_2(object):
|
||
|
|
def test_import2(self):
|
||
|
|
import polytestmix.test3
|
||
|
|
polytestmix.install(compiler2, ['.2'])
|
||
|
|
polytestmix.install(compiler3, ['.3'])
|
||
|
|
assert(polytestmix.test3.result = "Success for test3")
|
||
|
|
|
||
|
|
class Test_Polymorph_Iterator(object):
|
||
|
|
''' The Django Compatibility test. '''
|
||
|
|
def test_iterator(self):
|
||
|
|
import polytestmix.test3
|
||
|
|
polytestmix.install(compiler2, ['.2'])
|
||
|
|
polytestmix.install(compiler3, ['.3'])
|
||
|
|
target_dir = os.path.join('.', 'polytestmix')
|
||
|
|
files = set([name for _, name, is_pkg in pkgutil.iter_modules([targetdir])
|
||
|
|
if not is_pkg and not name.startswith('_')])
|
||
|
|
assert(files == set(['test2.2', 'test3.3', 'test1.py']))
|
||
|
|
|
||
|
|
|