Marmelator/layout-parser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 289 Wiki Activity

RED-6009: add direct json output test

Browse Source
This commit is contained in:
Kilian Schuettler 2023-04-19 12:22:19 +02:00
parent e60eb0710e
commit c912d27d92
2 changed files with 39 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphJsonWritingTest.java Normal file
View File

@ -0,0 +1,38 @@
package com.knecon.fforesight.service.layoutparser.server.graph;
import java.io.File;
import java.io.FileOutputStream;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.iqser.red.commons.jackson.ObjectMapperFactory;
import com.knecon.fforesight.service.layoutparser.internal.api.data.DocumentData;
import com.knecon.fforesight.service.layoutparser.internal.api.graph.DocumentGraph;
import com.knecon.fforesight.service.layoutparser.internal.api.mapper.DocumentDataMapper;
import lombok.SneakyThrows;
public class DocumentGraphJsonWritingTest extends BuildDocumentGraphTest {
@Test
@Disabled
public void writeJsonForFileTest() {
writeJsons("files/S-Metolachlor_RAR_01_Volume_1_2018-09-06");
}
@SneakyThrows
private void writeJsons(String filename) {
DocumentGraph documentGraph = buildGraph(filename);
DocumentData documentData = DocumentDataMapper.toDocumentData(documentGraph);
ObjectMapper mapper = ObjectMapperFactory.create();
mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_structure", "json")), documentData.getTableOfContents());
mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_text", "json")), documentData.getTableOfContents());
mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_positions", "json")), documentData.getTableOfContents());
mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_pages", "json")), documentData.getTableOfContents());
}
}

2
layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java
View File

@ -33,7 +33,7 @@ public class DocumentGraphVisualizationTest extends BuildDocumentGraphTest {
@Disabled
public void visualizeRotatedTestDocument() {
String filename = "files/RotateTestFileWithImages";
String filename = "files/211";
visualizePdf(filename);
}