Skip to content

Commit

Permalink
Linting
Browse files Browse the repository at this point in the history
  • Loading branch information
tibor-reiss committed Sep 11, 2024
1 parent 9f7a524 commit 5120b1a
Showing 1 changed file with 12 additions and 4 deletions.
16 changes: 12 additions & 4 deletions integration/test_collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@
def get_numpy_vector(input_list: list) -> Any:
try:
import numpy as np

return np.array(input_list)
except ModuleNotFoundError:
return input_list
Expand Down Expand Up @@ -307,21 +308,28 @@ class TestInsertManyWithTypedDict(TypedDict):
[
(
[
DataObject(properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3])),
DataObject(
properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3])
),
],
False,
),
(
[
DataObject(properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3])),
DataObject(properties={"name": "some numpy two"}, vector=get_numpy_vector([11, 12, 13])),
DataObject(
properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3])
),
DataObject(
properties={"name": "some numpy two"}, vector=get_numpy_vector([11, 12, 13])
),
],
False,
),
(
[
DataObject(
properties={"name": "some numpy 2d"}, vector=get_numpy_vector([[1, 2, 3], [11, 12, 13]])
properties={"name": "some numpy 2d"},
vector=get_numpy_vector([[1, 2, 3], [11, 12, 13]]),
),
],
True,
Expand Down

0 comments on commit 5120b1a

Please sign in to comment.