From 077a94a4b4fcefdf0b9f945eaba7b99480397919 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Thu, 4 Apr 2024 12:50:44 +0200 Subject: [PATCH] Fix TextSplitterTests --- .../ai/transformer/splitter/TextSplitterTests.java | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/spring-ai-core/src/test/java/org/springframework/ai/transformer/splitter/TextSplitterTests.java b/spring-ai-core/src/test/java/org/springframework/ai/transformer/splitter/TextSplitterTests.java index 8e25c9e8b..6c149dd04 100644 --- a/spring-ai-core/src/test/java/org/springframework/ai/transformer/splitter/TextSplitterTests.java +++ b/spring-ai-core/src/test/java/org/springframework/ai/transformer/splitter/TextSplitterTests.java @@ -19,12 +19,12 @@ import java.util.ArrayList; import java.util.List; import java.util.Map; -import static org.assertj.core.api.Assertions.assertThat; import org.junit.jupiter.api.Test; import org.springframework.ai.document.DefaultContentFormatter; import org.springframework.ai.document.Document; -import org.springframework.ai.transformer.splitter.TextSplitter; + +import static org.assertj.core.api.Assertions.assertThat; /** * @author Christian Tzolov @@ -81,9 +81,9 @@ public class TextSplitterTests { // Verify that the same, merged metadata is copied to all chunks. assertThat(chunks.get(0).getMetadata()).isEqualTo(chunks.get(1).getMetadata()); - assertThat(chunks.get(0).getMetadata()).isEqualTo(chunks.get(2).getMetadata()); - assertThat(chunks.get(0).getMetadata()).isEqualTo(chunks.get(3).getMetadata()); - assertThat(chunks.get(0).getMetadata()).containsKeys("key1", "key2", "key3"); + assertThat(chunks.get(2).getMetadata()).isEqualTo(chunks.get(3).getMetadata()); + assertThat(chunks.get(0).getMetadata()).containsKeys("key1", "key2").doesNotContainKeys("key3"); + assertThat(chunks.get(2).getMetadata()).containsKeys("key2", "key3").doesNotContainKeys("key1"); // Verify that the content formatters are copied from the parents to the chunks. // doc1 -> chunk0, chunk1 and doc2 -> chunk2, chunk3