add __contains__

This commit is contained in:
Thomas Levine 2014-08-29 18:12:25 +00:00
parent 1bdd35ee84
commit 17bad827ae
2 changed files with 6 additions and 0 deletions

View File

@ -141,6 +141,9 @@ class Database(object):
set(self.metadata.tables.keys()) | set(self._tables.keys())
)
def __contains__(self, member):
return member in self.tables
def create_table(self, table_name, primary_id='id', primary_type='Integer'):
"""
Creates a new table. The new table will automatically have an `id` column

View File

@ -38,6 +38,9 @@ class DatabaseTestCase(unittest.TestCase):
def test_tables(self):
assert self.db.tables == ['weather'], self.db.tables
def test_contains(self):
assert 'weather' in self.db, self.db.tables
def test_create_table(self):
table = self.db['foo']
assert table.table.exists()