Seems it was that lock.

Browse Source
This commit is contained in:
Friedrich Lindenberg 2020-08-23 12:23:00 +02:00
parent 28109d6826
commit 3fe0ae10aa
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dataset/table.py
View File

@ -528,8 +528,8 @@ class Table(object):
self._threading_warn() self._threading_warn()
self.table.drop(self.db.executable, checkfirst=True) self.table.drop(self.db.executable, checkfirst=True)
self._table = None self._table = None
self._columns = None
self.db._tables.pop(self.name, None) self.db._tables.pop(self.name, None)
self._flush_metadata()
def has_index(self, columns): def has_index(self, columns):
"""Check if an index exists to cover the given ``columns``.""" """Check if an index exists to cover the given ``columns``."""

16
test/test_dataset.py
View File

@ -2,7 +2,7 @@ import os
import unittest import unittest
from datetime import datetime from datetime import datetime
from collections import OrderedDict from collections import OrderedDict
from sqlalchemy import FLOAT, TEXT, BIGINT from sqlalchemy import TEXT, BIGINT
from sqlalchemy.exc import IntegrityError, SQLAlchemyError, ArgumentError from sqlalchemy.exc import IntegrityError, SQLAlchemyError, ArgumentError
from dataset import connect, chunked from dataset import connect, chunked
@ -153,10 +153,12 @@ class TableTestCase(unittest.TestCase):
def setUp(self): def setUp(self):
self.db = connect() self.db = connect()
self.tbl = self.db["weather"] self.tbl = self.db["weather"]
self.tbl.delete()
for row in TEST_DATA: for row in TEST_DATA:
self.tbl.insert(row) self.tbl.insert(row)
def tearDown(self):
self.tbl.drop()
def test_insert(self): def test_insert(self):
assert len(self.tbl) == len(TEST_DATA), len(self.tbl) assert len(self.tbl) == len(TEST_DATA), len(self.tbl)
last_id = self.tbl.insert( last_id = self.tbl.insert(
@ -386,7 +388,7 @@ class TableTestCase(unittest.TestCase):
def test_insert_many(self): def test_insert_many(self):
data = TEST_DATA * 100 data = TEST_DATA * 100
self.tbl.insert_many(data, chunk_size=13) self.tbl.insert_many(data, chunk_size=13)
assert len(self.tbl) == len(data) assert len(self.tbl) == len(data) + 6, (len(self.tbl), len(data))
def test_chunked_insert(self): def test_chunked_insert(self):
data = TEST_DATA * 100 data = TEST_DATA * 100
@ -499,16 +501,18 @@ class TableTestCase(unittest.TestCase):
def test_create_column(self): def test_create_column(self):
tbl = self.tbl tbl = self.tbl
tbl.create_column("foo", FLOAT) flt = self.db.types.float
tbl.create_column("foo", flt)
assert "foo" in tbl.table.c, tbl.table.c assert "foo" in tbl.table.c, tbl.table.c
assert isinstance(tbl.table.c["foo"].type, FLOAT), tbl.table.c["foo"].type assert isinstance(tbl.table.c["foo"].type, flt), tbl.table.c["foo"].type
assert "foo" in tbl.columns, tbl.columns assert "foo" in tbl.columns, tbl.columns
def test_ensure_column(self): def test_ensure_column(self):
tbl = self.tbl tbl = self.tbl
flt = self.db.types.float
tbl.create_column_by_example("foo", 0.1) tbl.create_column_by_example("foo", 0.1)
assert "foo" in tbl.table.c, tbl.table.c assert "foo" in tbl.table.c, tbl.table.c
assert isinstance(tbl.table.c["foo"].type, FLOAT), tbl.table.c["bar"].type assert isinstance(tbl.table.c["foo"].type, flt), tbl.table.c["bar"].type
tbl.create_column_by_example("bar", 1) tbl.create_column_by_example("bar", 1)
assert "bar" in tbl.table.c, tbl.table.c assert "bar" in tbl.table.c, tbl.table.c
assert isinstance(tbl.table.c["bar"].type, BIGINT), tbl.table.c["bar"].type assert isinstance(tbl.table.c["bar"].type, BIGINT), tbl.table.c["bar"].type