Skip to content

Commit 27c6be1

Browse files
committed
Add fixture enable_feature_32bit_dims
Ensures that dj.blob.use_32bit_dims is turned off even if test_insert_longblob fails.
1 parent bf17e75 commit 27c6be1

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

tests/test_blob.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import pytest
12
import datajoint as dj
23
import timeit
34
import numpy as np
@@ -10,6 +11,13 @@
1011
from .schema import Longblob
1112

1213

14+
@pytest.fixture
15+
def enable_feature_32bit_dims():
16+
dj.blob.use_32bit_dims = True
17+
yield
18+
dj.blob.use_32bit_dims = False
19+
20+
1321
def test_pack():
1422
for x in (
1523
32,
@@ -180,6 +188,8 @@ def test_insert_longblob(schema_any):
180188
assert (Longblob & "id=1").fetch1()["data"].all() == query_mym_blob["data"].all()
181189
(Longblob & "id=1").delete()
182190

191+
192+
def test_insert_longblob_32bit(schema_any, enable_feature_32bit_dims):
183193
query_32_blob = (
184194
"INSERT INTO djtest_test1.longblob (id, data) VALUES (1, "
185195
"X'6D596D00530200000001000000010000000400000068697473007369646573007461736B73007374"
@@ -190,7 +200,6 @@ def test_insert_longblob(schema_any):
190200
"00000041020000000100000008000000040000000000000053007400610067006500200031003000')"
191201
)
192202
dj.conn().query(query_32_blob).fetchall()
193-
dj.blob.use_32bit_dims = True
194203
fetched = (Longblob & "id=1").fetch1()
195204
expected = {
196205
"id": 1,
@@ -211,7 +220,6 @@ def test_insert_longblob(schema_any):
211220
assert fetched["id"] == expected["id"]
212221
assert np.array_equal(fetched["data"], expected["data"])
213222
(Longblob & "id=1").delete()
214-
dj.blob.use_32bit_dims = False
215223

216224

217225
def test_datetime_serialization_speed():

0 commit comments

Comments
 (0)