Skip to content

Commit 93859bc

Browse files
committed
Fix tobytes calls
1 parent 186c1c2 commit 93859bc

File tree

1 file changed

+18
-18
lines changed

1 file changed

+18
-18
lines changed

singlestoredb/tests/test_ext_func_data.py

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ class TestRowdat1(unittest.TestCase):
269269
def test_numpy_accel(self):
270270
dump_res = rowdat_1._dump_numpy_accel(
271271
col_types, numpy_row_ids, numpy_data,
272-
).tobytes()
272+
)
273273
load_res = rowdat_1._load_numpy_accel(col_spec, dump_res)
274274

275275
ids = load_res[0]
@@ -294,7 +294,7 @@ def test_numpy_accel(self):
294294
def test_numpy(self):
295295
dump_res = rowdat_1._dump_numpy(
296296
col_types, numpy_row_ids, numpy_data,
297-
).tobytes()
297+
)
298298
load_res = rowdat_1._load_numpy(col_spec, dump_res)
299299

300300
ids = load_res[0]
@@ -387,7 +387,7 @@ def test_numpy_accel_limits(self, name, dtype, data, res):
387387
with self.assertRaises(res, msg=f'Expected {res} for {data} in {dtype}'):
388388
rowdat_1._dump_numpy_accel(
389389
[dtype], numpy_row_ids, [(arr, None)],
390-
).tobytes()
390+
)
391391

392392
# Pure Python
393393
if 'mediumint exceeds' in name:
@@ -396,21 +396,21 @@ def test_numpy_accel_limits(self, name, dtype, data, res):
396396
with self.assertRaises(res, msg=f'Expected {res} for {data} in {dtype}'):
397397
rowdat_1._dump_numpy(
398398
[dtype], numpy_row_ids, [(arr, None)],
399-
).tobytes()
399+
)
400400

401401
else:
402402
# Accelerated
403403
dump_res = rowdat_1._dump_numpy_accel(
404404
[dtype], numpy_row_ids, [(arr, None)],
405-
).tobytes()
405+
)
406406
load_res = rowdat_1._load_numpy_accel([('x', dtype)], dump_res)
407407
assert load_res[1][0][0] == res, \
408408
f'Expected {res} for {data}, but got {load_res[1][0][0]} in {dtype}'
409409

410410
# Pure Python
411411
dump_res = rowdat_1._dump_numpy(
412412
[dtype], numpy_row_ids, [(arr, None)],
413-
).tobytes()
413+
)
414414
load_res = rowdat_1._load_numpy([('x', dtype)], dump_res)
415415
assert load_res[1][0][0] == res, \
416416
f'Expected {res} for {data}, but got {load_res[1][0][0]} in {dtype}'
@@ -788,7 +788,7 @@ def test_numpy_accel_casts(self, name, dtype, data, res):
788788
# Accelerated
789789
dump_res = rowdat_1._dump_numpy_accel(
790790
[dtype], numpy_row_ids, [(data, None)],
791-
).tobytes()
791+
)
792792
load_res = rowdat_1._load_numpy_accel([('x', dtype)], dump_res)
793793

794794
if name == 'double from float32':
@@ -800,7 +800,7 @@ def test_numpy_accel_casts(self, name, dtype, data, res):
800800
# Pure Python
801801
dump_res = rowdat_1._dump_numpy(
802802
[dtype], numpy_row_ids, [(data, None)],
803-
).tobytes()
803+
)
804804
load_res = rowdat_1._load_numpy([('x', dtype)], dump_res)
805805

806806
if name == 'double from float32':
@@ -812,7 +812,7 @@ def test_numpy_accel_casts(self, name, dtype, data, res):
812812
def test_python(self):
813813
dump_res = rowdat_1._dump(
814814
col_types, py_row_ids, py_col_data,
815-
).tobytes()
815+
)
816816
load_res = rowdat_1._load(col_spec, dump_res)
817817

818818
ids = load_res[0]
@@ -824,7 +824,7 @@ def test_python(self):
824824
def test_python_accel(self):
825825
dump_res = rowdat_1._dump_accel(
826826
col_types, py_row_ids, py_col_data,
827-
).tobytes()
827+
)
828828
load_res = rowdat_1._load_accel(col_spec, dump_res)
829829

830830
ids = load_res[0]
@@ -836,7 +836,7 @@ def test_python_accel(self):
836836
def test_polars(self):
837837
dump_res = rowdat_1._dump_polars(
838838
col_types, polars_row_ids, polars_data,
839-
).tobytes()
839+
)
840840
load_res = rowdat_1._load_polars(col_spec, dump_res)
841841

842842
ids = load_res[0]
@@ -861,7 +861,7 @@ def test_polars(self):
861861
def test_polars_accel(self):
862862
dump_res = rowdat_1._dump_polars_accel(
863863
col_types, polars_row_ids, polars_data,
864-
).tobytes()
864+
)
865865
load_res = rowdat_1._load_polars_accel(col_spec, dump_res)
866866

867867
ids = load_res[0]
@@ -886,7 +886,7 @@ def test_polars_accel(self):
886886
def test_pandas(self):
887887
dump_res = rowdat_1._dump_pandas(
888888
col_types, pandas_row_ids, pandas_data,
889-
).tobytes()
889+
)
890890
load_res = rowdat_1._load_pandas(col_spec, dump_res)
891891

892892
ids = load_res[0]
@@ -911,7 +911,7 @@ def test_pandas(self):
911911
def test_pandas_accel(self):
912912
dump_res = rowdat_1._dump_pandas_accel(
913913
col_types, pandas_row_ids, pandas_data,
914-
).tobytes()
914+
)
915915
load_res = rowdat_1._load_pandas_accel(col_spec, dump_res)
916916

917917
ids = load_res[0]
@@ -936,7 +936,7 @@ def test_pandas_accel(self):
936936
def test_pyarrow(self):
937937
dump_res = rowdat_1._dump_arrow(
938938
col_types, pyarrow_row_ids, pyarrow_data,
939-
).tobytes()
939+
)
940940
load_res = rowdat_1._load_arrow(col_spec, dump_res)
941941

942942
ids = load_res[0]
@@ -961,7 +961,7 @@ def test_pyarrow(self):
961961
def test_pyarrow_accel(self):
962962
dump_res = rowdat_1._dump_arrow_accel(
963963
col_types, pyarrow_row_ids, pyarrow_data,
964-
).tobytes()
964+
)
965965
load_res = rowdat_1._load_arrow_accel(col_spec, dump_res)
966966

967967
ids = load_res[0]
@@ -1053,7 +1053,7 @@ def test_polars(self):
10531053
def test_pandas(self):
10541054
dump_res = rowdat_1._dump_pandas(
10551055
col_types, pandas_row_ids, pandas_data,
1056-
).tobytes()
1056+
)
10571057
load_res = rowdat_1._load_pandas(col_spec, dump_res)
10581058

10591059
ids = load_res[0]
@@ -1078,7 +1078,7 @@ def test_pandas(self):
10781078
def test_pyarrow(self):
10791079
dump_res = rowdat_1._dump_arrow(
10801080
col_types, pyarrow_row_ids, pyarrow_data,
1081-
).tobytes()
1081+
)
10821082
load_res = rowdat_1._load_arrow(col_spec, dump_res)
10831083

10841084
ids = load_res[0]

0 commit comments

Comments
 (0)