diff --git a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/graph/ExtractRelationFromEntityTest.java b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/graph/ExtractRelationFromEntityTest.java index 706089e4d..109458355 100644 --- a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/graph/ExtractRelationFromEntityTest.java +++ b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/graph/ExtractRelationFromEntityTest.java @@ -27,7 +27,6 @@ import eu.dnetlib.dhp.oa.graph.dump.community.CommunityMap; import eu.dnetlib.dhp.schema.dump.oaf.graph.Relation; import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService; -@Disabled public class ExtractRelationFromEntityTest { private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper(); @@ -39,34 +38,6 @@ public class ExtractRelationFromEntityTest { private static final Logger log = LoggerFactory .getLogger(ExtractRelationFromEntityTest.class); - private static CommunityMap map = new CommunityMap(); - - static { - map.put("egi", "EGI Federation"); - map.put("fet-fp7", "FET FP7"); - map.put("fet-h2020", "FET H2020"); - map.put("clarin", "CLARIN"); - map.put("fam", "Fisheries and Aquaculture Management"); - map.put("ni", "Neuroinformatics"); - map.put("mes", "European Marine Scinece"); - map.put("instruct", "Instruct-Eric"); - map.put("rda", "Research Data Alliance"); - map.put("elixir-gr", "ELIXIR GR"); - map.put("aginfra", "Agricultural and Food Sciences"); - map.put("dariah", "DARIAH EU"); - map.put("risis", "RISI"); - map.put("ee", "SDSN - Greece"); - map.put("oa-pg", "EC Post-Grant Open Access Pilot"); - map.put("beopen", "Transport Research"); - map.put("euromarine", "Euromarine"); - map.put("ifremer", "Ifremer"); - map.put("dh-ch", "Digital Humanities and Cultural Heritage"); - map.put("science-innovation-policy", "Science and Innovation Policy Studies"); - map.put("covid-19", "COVID-19"); - map.put("enrmaps", "Energy Research"); - map.put("epos", "EPOS"); - - } @BeforeAll public static void beforeAll() throws IOException { @@ -97,43 +68,43 @@ public class ExtractRelationFromEntityTest { spark.stop(); } -// @Test -// public void test1() { -// -// final String sourcePath = getClass() -// .getResource("/eu/dnetlib/dhp/oa/graph/dump/resultDump/singelRecord_pub.json") -// .getPath(); -// -// final String communityMapPath = getClass() -// .getResource("/eu/dnetlib/dhp/oa/graph/dump/communityMapPath/communitymap.json") -// .getPath(); -// -// Extractor ex = new Extractor(); -// ex -// .run( -// false, sourcePath, workingDir.toString() + "/relation", -// // eu.dnetlib.dhp.schema.oaf.Publication.class, communityMapPath); -// eu.dnetlib.dhp.schema.oaf.Publication.class, map); -// -// final JavaSparkContext sc = JavaSparkContext.fromSparkContext(spark.sparkContext()); -// -// JavaRDD tmp = sc -// .textFile(workingDir.toString() + "/relation") -// .map(item -> OBJECT_MAPPER.readValue(item, Relation.class)); -// -// org.apache.spark.sql.Dataset verificationDataset = spark -// .createDataset(tmp.rdd(), Encoders.bean(Relation.class)); -// -// Assertions -// .assertEquals( -// 9, -// verificationDataset.filter("source.id = '50|dedup_wf_001::15270b996fa8fd2fb5723daeab3685c3'").count()); -// -// Assertions -// .assertEquals( -// 9, -// verificationDataset.filter("source.id = '50|dedup_wf_001::15270b996fa8fd2fb5723daxab3685c3'").count()); -// -// } + @Test + public void test1() { + + final String sourcePath = getClass() + .getResource("/eu/dnetlib/dhp/oa/graph/dump/resultDump/singelRecord_pub.json") + .getPath(); + + final String communityMapPath = getClass() + .getResource("/eu/dnetlib/dhp/oa/graph/dump/communityMapPath/communitymap.json") + .getPath(); + + Extractor ex = new Extractor(); + ex + .run( + false, sourcePath, workingDir.toString() + "/relation", + // eu.dnetlib.dhp.schema.oaf.Publication.class, communityMapPath); + eu.dnetlib.dhp.schema.oaf.Publication.class, communityMapPath); + + final JavaSparkContext sc = JavaSparkContext.fromSparkContext(spark.sparkContext()); + + JavaRDD tmp = sc + .textFile(workingDir.toString() + "/relation") + .map(item -> OBJECT_MAPPER.readValue(item, Relation.class)); + + org.apache.spark.sql.Dataset verificationDataset = spark + .createDataset(tmp.rdd(), Encoders.bean(Relation.class)); + + Assertions + .assertEquals( + 9, + verificationDataset.filter("source.id = '50|dedup_wf_001::15270b996fa8fd2fb5723daeab3685c3'").count()); + + Assertions + .assertEquals( + 9, + verificationDataset.filter("source.id = '50|dedup_wf_001::15270b996fa8fd2fb5723daxab3685c3'").count()); + + } }