mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-06-04 06:48:27 -04:00
Mike Bayer
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""Enhance unittest and instrument SQLAlchemy classes for testing.
|
|
|
|
Load after sqlalchemy imports to use instrumented stand-ins like Table.
|
|
"""
|
|
|
|
import sys
|
|
import testlib.config
|
|
from testlib.schema import Table, Column
|
|
from testlib.orm import mapper
|
|
import testlib.testing as testing
|
|
from testlib.testing import \
|
|
AssertsCompiledSQL, \
|
|
AssertsExecutionResults, \
|
|
ComparesTables, \
|
|
ORMTest, \
|
|
TestBase, \
|
|
rowset
|
|
import testlib.profiling as profiling
|
|
import testlib.engines as engines
|
|
import testlib.requires as requires
|
|
from testlib.compat import set, frozenset, sorted, _function_named
|
|
|
|
|
|
__all__ = ('testing',
|
|
'mapper',
|
|
'Table', 'Column',
|
|
'rowset',
|
|
'TestBase', 'AssertsExecutionResults', 'ORMTest',
|
|
'AssertsCompiledSQL', 'ComparesTables',
|
|
'profiling', 'engines',
|
|
'set', 'frozenset', 'sorted', '_function_named')
|
|
|
|
|
|
testing.requires = requires
|
|
|
|
sys.modules['testlib.sa'] = sa = testing.CompositeModule(
|
|
'testlib.sa', 'sqlalchemy', 'testlib.schema', orm=testing.CompositeModule(
|
|
'testlib.sa.orm', 'sqlalchemy.orm', 'testlib.orm'))
|
|
sys.modules['testlib.sa.orm'] = sa.orm
|