Fix assertions

This commit is contained in:
Eddú Meléndez
2024-07-16 15:47:44 -05:00
committed by Christian Tzolov
parent f783796934
commit 0a42bf01f3
2 changed files with 4 additions and 6 deletions

View File

@@ -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);
}

View File

@@ -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);
});
}