1
0
Fork 0

[graph cleaning] WIP: refactoring of the cleaning stages, unit tests

This commit is contained in:
Claudio Atzori 2023-03-23 15:03:26 +01:00
parent 488d9a5eaa
commit 90e61a8aba
2 changed files with 6 additions and 5 deletions

View File

@ -237,7 +237,7 @@ public class CleanGraphSparkJobTest {
final String id = "50|CSC_________::2250a70c903c6ac6e4c01438259e9375"; final String id = "50|CSC_________::2250a70c903c6ac6e4c01438259e9375";
Publication p_in = read(spark, graphOutputPath.toString() + "/publication", Publication.class) Publication p_in = read(spark, graphInputPath.toString() + "/publication", Publication.class)
.filter(String.format("id = '%s'", id)) .filter(String.format("id = '%s'", id))
.first(); .first();
@ -437,7 +437,7 @@ public class CleanGraphSparkJobTest {
Dataset<Publication> pubs = read(spark, graphOutputPath.toString() + "/publication", Publication.class) Dataset<Publication> pubs = read(spark, graphOutputPath.toString() + "/publication", Publication.class)
.filter((FilterFunction<Publication>) p1 -> StringUtils.endsWith(p1.getId(), "_ctx")); .filter((FilterFunction<Publication>) p1 -> StringUtils.endsWith(p1.getId(), "_ctx"));
Assertions.assertEquals(7, pubs.count()); assertEquals(7, pubs.count());
// original result with sobigdata context and gcube as starting string in the main title for the publication // original result with sobigdata context and gcube as starting string in the main title for the publication
assertEquals( assertEquals(
@ -730,7 +730,7 @@ public class CleanGraphSparkJobTest {
final Dataset<Publication> pubs_out = read(spark, graphOutputPath.toString() + "/publication", Publication.class) final Dataset<Publication> pubs_out = read(spark, graphOutputPath.toString() + "/publication", Publication.class)
.filter((FilterFunction<Publication>) p -> StringUtils.endsWith(p.getId(), "_country")); .filter((FilterFunction<Publication>) p -> StringUtils.endsWith(p.getId(), "_country"));
Assertions.assertEquals(7, pubs_out.count()); assertEquals(8, pubs_out.count());
// original result with NL country and doi starting with Mendely prefix, but not collectedfrom NARCIS // original result with NL country and doi starting with Mendely prefix, but not collectedfrom NARCIS
assertEquals( assertEquals(
@ -765,7 +765,7 @@ public class CleanGraphSparkJobTest {
assertEquals( assertEquals(
0, 0,
pubs_out pubs_out
.filter((FilterFunction<Publication>) p -> p.getId().equals("50|DansKnawCris::3c81248c335f0aa07e06817e_country")) .filter((FilterFunction<Publication>) p -> p.getId().equals("50|DansKnawCris::3c81248c335f0aa07e06817d_country"))
.first() .first()
.getCountry() .getCountry()
.size()); .size());