1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.giraph.examples.block_app;
20
21 import static org.junit.Assert.assertEquals;
22 import static org.junit.Assert.assertTrue;
23
24 import org.apache.giraph.BspCase;
25 import org.apache.giraph.block_app.framework.BlockUtils;
26 import org.apache.giraph.conf.GiraphConfiguration;
27 import org.apache.giraph.examples.SimplePageRankComputation.SimplePageRankVertexInputFormat;
28 import org.apache.giraph.job.GiraphJob;
29 import org.junit.Test;
30
31
32
33
34 public class
35 TestMigrationBspBasic extends BspCase {
36
37 public TestMigrationBspBasic() {
38 super(TestMigrationBspBasic.class.getName());
39 }
40
41
42
43
44
45 @Test
46 public void testBspMigrationToBlocksFramework() throws Exception {
47 GiraphConfiguration conf = new GiraphConfiguration();
48 conf.setVertexInputFormatClass(SimplePageRankVertexInputFormat.class);
49 BlockUtils.setAndInitBlockFactoryClass(
50 conf, SimpleMigrationMasterBlockFactory.class);
51 GiraphJob job = prepareJob(getCallingMethodName(), conf);
52 assertTrue(job.run(true));
53 if (!runningInDistributedMode()) {
54 double finalSum =
55 SimpleMigrationMasterBlockFactory.SimpleMigrationMasterWorkerContext.getFinalSum();
56 System.out.println("testBspMasterCompute: finalSum=" + finalSum);
57 assertEquals(32.5, finalSum, 0d);
58 }
59 }
60 }