Skip to content

Commit 1ce5481

Browse files
chore(deps): bump the maven group across 1 directory with 2 updates (#17)
* chore(deps): bump the maven group across 1 directory with 2 updates Bumps the maven group with 2 updates in the / directory: [com.google.guava:guava](https://github.com/google/guava) and [junit:junit](https://github.com/junit-team/junit4). Updates `com.google.guava:guava` from 30.1-jre to 32.0.0-jre - [Release notes](https://github.com/google/guava/releases) - [Commits](https://github.com/google/guava/commits) Updates `junit:junit` from 4.11 to 4.13.1 - [Release notes](https://github.com/junit-team/junit4/releases) - [Changelog](https://github.com/junit-team/junit4/blob/main/doc/ReleaseNotes4.11.md) - [Commits](junit-team/junit4@r4.11...r4.13.1) --- updated-dependencies: - dependency-name: com.google.guava:guava dependency-type: direct:production dependency-group: maven-security-group - dependency-name: junit:junit dependency-type: direct:development dependency-group: maven-security-group ... Signed-off-by: dependabot[bot] <[email protected]> * test: assert test results --------- Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Anush008 <[email protected]>
1 parent 0c98219 commit 1ce5481

File tree

2 files changed

+12
-11
lines changed

2 files changed

+12
-11
lines changed

pom.xml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
<dependency>
4545
<groupId>com.google.guava</groupId>
4646
<artifactId>guava</artifactId>
47-
<version>30.1-jre</version>
47+
<version>32.0.0-jre</version>
4848
</dependency>
4949
<dependency>
5050
<groupId>io.grpc</groupId>
@@ -75,7 +75,7 @@
7575
<dependency>
7676
<groupId>junit</groupId>
7777
<artifactId>junit</artifactId>
78-
<version>4.11</version>
78+
<version>4.13.1</version>
7979
<scope>test</scope>
8080
</dependency>
8181
<dependency>

src/test/python/test_qdrant_ingest.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def test_upsert_unnamed_vectors(qdrant: Qdrant, spark_session: SparkSession):
2020
"embedding_field", "dense_vector"
2121
).mode("append").option("schema", df.schema.json()).save()
2222

23-
qdrant.client.count(qdrant.collection_name) == df.count()
23+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
2424

2525

2626
def test_upsert_named_vectors(qdrant: Qdrant, spark_session: SparkSession):
@@ -38,7 +38,8 @@ def test_upsert_named_vectors(qdrant: Qdrant, spark_session: SparkSession):
3838
"append"
3939
).save()
4040

41-
qdrant.client.count(qdrant.collection_name) == df.count()
41+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
42+
4243

4344

4445
def test_upsert_multiple_named_dense_vectors(
@@ -58,7 +59,7 @@ def test_upsert_multiple_named_dense_vectors(
5859
"schema", df.schema.json()
5960
).mode("append").save()
6061

61-
qdrant.client.count(qdrant.collection_name) == df.count()
62+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
6263

6364

6465
def test_upsert_sparse_vectors(qdrant: Qdrant, spark_session: SparkSession):
@@ -76,7 +77,7 @@ def test_upsert_sparse_vectors(qdrant: Qdrant, spark_session: SparkSession):
7677
"sparse_vector_names", "sparse"
7778
).option("schema", df.schema.json()).mode("append").save()
7879

79-
qdrant.client.count(qdrant.collection_name) == df.count()
80+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
8081

8182

8283
def test_upsert_multiple_sparse_vectors(qdrant: Qdrant, spark_session: SparkSession):
@@ -94,7 +95,7 @@ def test_upsert_multiple_sparse_vectors(qdrant: Qdrant, spark_session: SparkSess
9495
"sparse_vector_names", "sparse,another_sparse"
9596
).option("schema", df.schema.json()).mode("append").save()
9697

97-
qdrant.client.count(qdrant.collection_name) == df.count()
98+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
9899

99100

100101
def test_upsert_sparse_named_dense_vectors(qdrant: Qdrant, spark_session: SparkSession):
@@ -114,7 +115,7 @@ def test_upsert_sparse_named_dense_vectors(qdrant: Qdrant, spark_session: SparkS
114115
"sparse_vector_names", "sparse"
115116
).option("schema", df.schema.json()).mode("append").save()
116117

117-
qdrant.client.count(qdrant.collection_name) == df.count()
118+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
118119

119120

120121
def test_upsert_sparse_unnamed_dense_vectors(
@@ -136,7 +137,7 @@ def test_upsert_sparse_unnamed_dense_vectors(
136137
"append"
137138
).save()
138139

139-
qdrant.client.count(qdrant.collection_name) == df.count()
140+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
140141

141142

142143
def test_upsert_multiple_sparse_dense_vectors(
@@ -158,7 +159,7 @@ def test_upsert_multiple_sparse_dense_vectors(
158159
"sparse_vector_names", "sparse,another_sparse"
159160
).option("schema", df.schema.json()).mode("append").save()
160161

161-
qdrant.client.count(qdrant.collection_name) == df.count()
162+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
162163

163164

164165
# Test an upsert without vectors. All the dataframe fields will be treated as payload
@@ -175,7 +176,7 @@ def test_upsert_without_vectors(qdrant: Qdrant, spark_session: SparkSession):
175176
"schema", df.schema.json()
176177
).mode("append").save()
177178

178-
qdrant.client.count(qdrant.collection_name) == df.count()
179+
assert qdrant.client.count(qdrant.collection_name).count == df.count(), "Uploaded points count is not equal to the dataframe count"
179180

180181

181182
def test_custom_id_field(qdrant: Qdrant, spark_session: SparkSession):

0 commit comments

Comments
 (0)