From 0a42bf01f36c265972d2032e4488204edcd3f900 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Edd=C3=BA=20Mel=C3=A9ndez?= Date: Tue, 16 Jul 2024 15:47:44 -0500 Subject: [PATCH] Fix assertions --- .../java/org/springframework/ai/chat/ChatBuilderTests.java | 4 ++-- .../springframework/ai/vectorstore/RedisVectorStoreIT.java | 6 ++---- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/spring-ai-core/src/test/java/org/springframework/ai/chat/ChatBuilderTests.java b/spring-ai-core/src/test/java/org/springframework/ai/chat/ChatBuilderTests.java index 19ece24f8..da9db9e79 100644 --- a/spring-ai-core/src/test/java/org/springframework/ai/chat/ChatBuilderTests.java +++ b/spring-ai-core/src/test/java/org/springframework/ai/chat/ChatBuilderTests.java @@ -97,12 +97,12 @@ public class ChatBuilderTests { // Callback Functions assertThat(options.getFunctionCallbacks()).isNotNull(); assertThat(options.getFunctionCallbacks().size()).isEqualTo(1); - assertThat(options.getFunctionCallbacks().contains(cb)); + assertThat(options.getFunctionCallbacks()).contains(cb); // Functions assertThat(options.getFunctions()).isNotNull(); assertThat(options.getFunctions().size()).isEqualTo(1); - assertThat(options.getFunctions().contains(func)); + assertThat(options.getFunctions()).contains(func); } diff --git a/vector-stores/spring-ai-redis-store/src/test/java/org/springframework/ai/vectorstore/RedisVectorStoreIT.java b/vector-stores/spring-ai-redis-store/src/test/java/org/springframework/ai/vectorstore/RedisVectorStoreIT.java index 735a45176..c766c1af4 100644 --- a/vector-stores/spring-ai-redis-store/src/test/java/org/springframework/ai/vectorstore/RedisVectorStoreIT.java +++ b/vector-stores/spring-ai-redis-store/src/test/java/org/springframework/ai/vectorstore/RedisVectorStoreIT.java @@ -78,10 +78,8 @@ class RedisVectorStoreIT { @Test void ensureIndexGetsCreated() { this.contextRunner.run(context -> { - assertThat(context.getBean(RedisVectorStore.class) - .getJedis() - .ftList() - .contains(RedisVectorStore.DEFAULT_INDEX_NAME)); + assertThat(context.getBean(RedisVectorStore.class).getJedis().ftList()) + .contains(RedisVectorStore.DEFAULT_INDEX_NAME); }); }