diff --git a/dataset/__init__.py b/dataset/__init__.py index dbc465f..0a1556c 100644 --- a/dataset/__init__.py +++ b/dataset/__init__.py @@ -16,8 +16,7 @@ __all__ = ['Database', 'Table', 'freeze', 'connect'] def connect(url=None, schema=None, reflect_metadata=True, engine_kwargs=None, reflect_views=True, row_type=row_type): - """ - Opens a new connection to a database. + """ Opens a new connection to a database. *url* can be any valid `SQLAlchemy engine URL`_. If *url* is not defined it will try to use *DATABASE_URL* from environment variable. Returns an @@ -28,8 +27,8 @@ def connect(url=None, schema=None, reflect_metadata=True, engine_kwargs=None, loaded. Additionally, *engine_kwargs* will be directly passed to SQLAlchemy, e.g. set *engine_kwargs={'pool_recycle': 3600}* will avoid `DB connection timeout`_. Set *row_type* to an alternate dict-like class to - change the type of container rows are stored in. - :: + change the type of container rows are stored in.:: + db = dataset.connect('sqlite:///factbook.db') .. _SQLAlchemy Engine URL: http://docs.sqlalchemy.org/en/latest/core/engines.html#sqlalchemy.create_engine diff --git a/test/test_freeze.py b/test/test_freeze.py index 16f640e..09c8b98 100644 --- a/test/test_freeze.py +++ b/test/test_freeze.py @@ -12,7 +12,7 @@ from dataset import connect from dataset.freeze.app import freeze from dataset.freeze.format.fcsv import value_to_str -from sample_data import TEST_DATA +from .sample_data import TEST_DATA class FreezeTestCase(unittest.TestCase): diff --git a/test/test_freeze_app.py b/test/test_freeze_app.py index 4a62acd..f9bee0d 100644 --- a/test/test_freeze_app.py +++ b/test/test_freeze_app.py @@ -14,7 +14,7 @@ from dataset import connect from dataset.util import FreezeException from dataset.freeze.config import Configuration, Export from dataset.freeze.app import create_parser, freeze_with_config, freeze_export -from sample_data import TEST_DATA +from .sample_data import TEST_DATA class FreezeAppTestCase(unittest.TestCase): diff --git a/test/test_persistence.py b/test/test_persistence.py index 92baf2f..7964543 100644 --- a/test/test_persistence.py +++ b/test/test_persistence.py @@ -12,7 +12,7 @@ from sqlalchemy.exc import IntegrityError, SQLAlchemyError from dataset import connect from dataset.util import DatasetException -from sample_data import TEST_DATA, TEST_CITY_1 +from .sample_data import TEST_DATA, TEST_CITY_1 class DatabaseTestCase(unittest.TestCase):