7
7
import com .marklogic .client .row .RowManager ;
8
8
import com .marklogic .client .row .RowRecord ;
9
9
import com .marklogic .client .row .RowSet ;
10
+ import com .marklogic .junit5 .RequiresMarkLogic12 ;
10
11
import com .marklogic .junit5 .XmlNode ;
11
12
import com .marklogic .spark .AbstractIntegrationTest ;
12
13
import com .marklogic .spark .ConnectorException ;
17
18
import org .apache .spark .sql .SaveMode ;
18
19
import org .jdom2 .Namespace ;
19
20
import org .junit .jupiter .api .Test ;
21
+ import org .junit .jupiter .api .extension .ExtendWith ;
20
22
21
23
import static org .junit .jupiter .api .Assertions .assertEquals ;
22
24
23
25
class AddEmbeddingsToXmlTest extends AbstractIntegrationTest {
24
26
25
27
private static final String TEST_EMBEDDING_FUNCTION_CLASS = "com.marklogic.spark.writer.embedding.MinilmEmbeddingModelFunction" ;
26
28
29
+ @ ExtendWith (RequiresMarkLogic12 .class )
27
30
@ Test
28
31
void embeddingsInChunkDocuments () {
29
32
readDocument ("/marklogic-docs/java-client-intro.xml" )
@@ -44,6 +47,7 @@ void embeddingsInChunkDocuments() {
44
47
verifyEachChunkIsReturnedByAVectorQuery ("xml_chunks" );
45
48
}
46
49
50
+ @ ExtendWith (RequiresMarkLogic12 .class )
47
51
@ Test
48
52
void embeddingsInSourceDocument () {
49
53
readDocument ("/marklogic-docs/java-client-intro.xml" )
@@ -62,6 +66,7 @@ void embeddingsInSourceDocument() {
62
66
verifyEachChunkIsReturnedByAVectorQuery ("xml_chunks" );
63
67
}
64
68
69
+ @ ExtendWith (RequiresMarkLogic12 .class )
65
70
@ Test
66
71
void sidecarWithNamespace () {
67
72
readDocument ("/marklogic-docs/namespaced-java-client-intro.xml" )
0 commit comments