Java Code Examples for org.apache.pig.pigunit.PigTest#assertOutput()

The following examples show how to use org.apache.pig.pigunit.PigTest#assertOutput() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: TestPigTest.java From spork with Apache License 2.0

6 votes

@Test
public void testNtoN() throws ParseException, IOException {
    String[] args = {
                    "n=3",
                    "reducers=1",
                    "input=top_queries_input_data.txt",
                    "output=top_3_queries",
    };
    test = new PigTest(PIG_SCRIPT, args);

    String[] output = {
                    "(yahoo,25)",
                    "(facebook,15)",
                    "(twitter,7)",
    };

    test.assertOutput("queries_limit", output);
}

Example 2

Source File: TestPigTest.java From spork with Apache License 2.0

6 votes

@Test
public void testImplicitNtoN() throws ParseException, IOException {
    String[] args = {
                    "n=3",
                    "reducers=1",
                    "input=top_queries_input_data.txt",
                    "output=top_3_queries",
    };
    test = new PigTest(PIG_SCRIPT, args);

    String[] output = {
                    "(yahoo,25)",
                    "(facebook,15)",
                    "(twitter,7)",
    };

    test.assertOutput(output);
}

Example 3

Source File: TestPigTest.java From spork with Apache License 2.0

6 votes

@Test
public void testOverride() throws ParseException, IOException {
    String[] args = {
                    "n=3",
                    "reducers=1",
                    "input=top_queries_input_data.txt",
                    "output=top_3_queries",
    };
    test = new PigTest(PIG_SCRIPT, args);

    test.override("queries_limit", "queries_limit = LIMIT queries_ordered 2;");

    String[] output = {
                    "(yahoo,25)",
                    "(facebook,15)",
    };

    test.assertOutput(output);
}

Example 4

Source File: TestPigTest.java From spork with Apache License 2.0

6 votes

@Test
public void testInlinePigScript() throws ParseException, IOException {
    String[] script = {
                    "data = LOAD 'top_queries_input_data.txt' AS (query:CHARARRAY, count:INT);",
                    "queries_group = GROUP data BY query PARALLEL 1;",
                    "queries_sum = FOREACH queries_group GENERATE group AS query, SUM(data.count) AS count;",
                    "queries_ordered = ORDER queries_sum BY count DESC PARALLEL 1;",
                    "queries_limit = LIMIT queries_ordered 3;",
                    "STORE queries_limit INTO 'top_3_queries';",
    };

    test = new PigTest(script);

    String[] output = {
                    "(yahoo,25)",
                    "(facebook,15)",
                    "(twitter,7)",
    };

    test.assertOutput(output);
}

Example 5

Source File: TestPigTest.java From spork with Apache License 2.0

6 votes

@Test
public void testWithUdf() throws ParseException, IOException {
    String[] script = {
                    // "REGISTER myIfNeeded.jar;",
                    "DEFINE TOKENIZE TOKENIZE();",
                    "data = LOAD 'top_queries_input_data.txt' AS (query:CHARARRAY, count:INT);",
                    "queries = FOREACH data GENERATE query, TOKENIZE(query) AS query_tokens;",
                    "queries_ordered = ORDER queries BY query DESC PARALLEL 1;",
                    "queries_limit = LIMIT queries_ordered 3;",
                    "STORE queries_limit INTO 'top_3_queries';",
    };

    test = new PigTest(script);

    String[] output = {
                    "(yahoo,{(yahoo)})",
                    "(yahoo,{(yahoo)})",
                    "(twitter,{(twitter)})",
    };

    test.assertOutput(output);
}

Example 6

Source File: TestPigTest.java From spork with Apache License 2.0

6 votes

@Ignore("Not ready yet")
@Test
public void testWithMock() throws ParseException, IOException {
    String[] args = {
                    "n=3",
                    "reducers=1",
                    "input=top_queries_input_data.txt",
                    "output=top_3_queries",
    };

    PigServer mockServer = null;
    Cluster mockCluster = null;

    test = new PigTest(PIG_SCRIPT, args, mockServer, mockCluster);

    test.assertOutput(new File("data/top_queries_expected_top_3.txt"));
}

Example 7

Source File: BagTests.java From datafu with Apache License 2.0

6 votes

@Test
public void unorderedPairsTest() throws Exception
{
  PigTest test = createPigTestFromString(unorderedPairsTest);

  String[] input = {
    "{(1),(2),(3),(4),(5)}"
  };

  String[] output = {
      "(1,2)",
      "(1,3)",
      "(1,4)",
      "(1,5)",
      "(2,3)",
      "(2,4)",
      "(2,5)",
      "(3,4)",
      "(3,5)",
      "(4,5)"
    };

  test.assertOutput("data",input,"data4",output);
}

Example 8

Source File: IntBoolConversionPigTests.java From datafu with Apache License 2.0

6 votes

@Test
public void intToBoolTest() throws Exception
{
  PigTest test = createPigTestFromString(intToBoolTest);
      
  String[] input = {
    "", // null
    "0",
    "1"
  };
  
  String[] output = {
      "(false)",
      "(false)",
      "(true)"
    };
  
  test.assertOutput("data",input,"data2",output);
}

Example 9

Source File: TestPigTest.java From spork with Apache License 2.0

5 votes

@Test
public void testTextInput() throws ParseException, IOException {
    String[] args = {
                    "n=3",
                    "reducers=1",
                    "input=top_queries_input_data.txt",
                    "output=top_3_queries",
    };
    test = new PigTest(PIG_SCRIPT, args);

    String[] input = {
                    "yahoo\t10",
                    "twitter\t7",
                    "facebook\t10",
                    "yahoo\t15",
                    "facebook\t5",
                    "a\t1",
                    "b\t2",
                    "c\t3",
                    "d\t4",
                    "e\t5",
    };

    String[] output = {
                    "(yahoo,25)",
                    "(facebook,15)",
                    "(twitter,7)",
    };

    test.assertOutput("data", input, "queries_limit", output);
}

Example 10

Source File: TestPigTest.java From spork with Apache License 2.0

5 votes

@Test
public void testDelimiter() throws ParseException, IOException {
    String[] args = {
                    "n=3",
                    "reducers=1",
                    "input=top_queries_input_data.txt",
                    "output=top_3_queries",
    };
    test = new PigTest(PIG_SCRIPT, args);

    String[] input = {
                    "yahoo,10",
                    "twitter,7",
                    "facebook,10",
                    "yahoo,15",
                    "facebook,5",
                    "a,1",
                    "b,2",
                    "c,3",
                    "d,4",
                    "e,5",
    };

    String[] output = {
                    "(yahoo,25)",
                    "(facebook,15)",
                    "(twitter,7)",
    };

    test.assertOutput("data", input, "queries_limit", output, ",");
}

Example 11

Source File: TestPigTest.java From spork with Apache License 2.0

5 votes

@Test
public void testSubset() throws ParseException, IOException {
    String[] args = {
                    "n=3",
                    "reducers=1",
                    "input=top_queries_input_data.txt",
                    "output=top_3_queries",
    };
    test = new PigTest(PIG_SCRIPT, args);

    String[] input = {
                    "yahoo\t10",
                    "twitter\t7",
                    "facebook\t10",
                    "yahoo\t15",
                    "facebook\t5",
                    "a\t1",
                    "b\t2",
                    "c\t3",
                    "d\t4",
                    "e\t5",
    };

    String[] output = {
                    "(yahoo,25)",
                    "(facebook,15)",
                    "(twitter,7)",
    };

    test.assertOutput("data", input, "queries_limit", output);
}

Example 12

Source File: TestPigTest.java From spork with Apache License 2.0

5 votes

@Test
public void testFileOutput() throws ParseException, IOException {
    String[] args = {
                    "n=3",
                    "reducers=1",
                    "input=top_queries_input_data.txt",
                    "output=top_3_queries",
    };
    test = new PigTest(PIG_SCRIPT, args);

    test.assertOutput(new File("test/data/pigunit/top_queries_expected_top_3.txt"));
}

Example 13

Source File: TestPigTest.java From spork with Apache License 2.0

5 votes

@Test
public void testArgFiles() throws ParseException, IOException {
    String[] argsFile = {
                    "test/data/pigunit/top_queries_params.txt"
    };

    test = new PigTest(PIG_SCRIPT, null, argsFile);

    test.assertOutput(new File("test/data/pigunit/top_queries_expected_top_3.txt"));
}

Example 14

Source File: SessionTests.java From datafu with Apache License 2.0

5 votes

@Test
public void sessionCountPageViewsTest() throws Exception
{
  PigTest test = createPigTestFromString(sessionCountPageViewsTest,
                               "TIME_WINDOW=30m");
      
  String[] input = {
    "1\t100\t2010-01-01T01:00:00Z",
    "1\t100\t2010-01-01T01:15:00Z",
    "1\t100\t2010-01-01T01:31:00Z",
    "1\t100\t2010-01-01T01:35:00Z",
    "1\t100\t2010-01-01T02:30:00Z",

    "1\t101\t2010-01-01T01:00:00Z",
    "1\t101\t2010-01-01T01:31:00Z",
    "1\t101\t2010-01-01T02:10:00Z",
    "1\t101\t2010-01-01T02:40:30Z",
    "1\t101\t2010-01-01T03:30:00Z",      

    "1\t102\t2010-01-01T01:00:00Z",
    "1\t102\t2010-01-01T01:01:00Z",
    "1\t102\t2010-01-01T01:02:00Z",
    "1\t102\t2010-01-01T01:10:00Z",
    "1\t102\t2010-01-01T01:15:00Z",
    "1\t102\t2010-01-01T01:25:00Z",
    "1\t102\t2010-01-01T01:30:00Z"
  };
  
  String[] output = {
      "(1,100,2)",
      "(1,101,5)",
      "(1,102,1)"
    };
  
  test.assertOutput("views",input,"view_counts",output);
}

Example 15

Source File: IntBoolConversionPigTests.java From datafu with Apache License 2.0

5 votes

@Test
public void intToBoolToIntTest() throws Exception
{
  PigTest test = createPigTestFromString(intToBoolToIntTest);
      
  String[] input = {
    "", // null
    "0",
    "1",
    "2",
    "-1",
    "-2",
    "0",
    ""
  };
  
  String[] output = {
      "(0)",
      "(0)",
      "(1)",
      "(1)",
      "(1)",
      "(1)",
      "(0)",
      "(0)"
    };
  
  test.assertOutput("data",input,"data3",output);
}

Example 16

Source File: UserAgentTest.java From datafu with Apache License 2.0

5 votes

@Test
public void userAgentTest() throws Exception
{
  PigTest test = createPigTestFromString(userAgentTest);

  String[] input = {
      "Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
      "Mozilla/5.0 (compatible; Konqueror/3.5; Linux; X11; de) KHTML/3.5.2 (like Gecko) Kubuntu 6.06 Dapper",
      "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:2.2a1pre) Gecko/20110331 Firefox/4.2a1pre Fennec/4.1a1pre",
      "Opera/9.00 (X11; Linux i686; U; en)",
      "Wget/1.10.2",
      "Opera/9.80 (Android; Linux; Opera Mobi/ADR-1012221546; U; pl) Presto/2.7.60 Version/10.5",
      "Mozilla/5.0 (Linux; U; Android 2.2; en-us; DROID2 Build/VZW) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1"
  };
  
  String[] output = {
      "(mobile)",
      "(desktop)",
      "(mobile)",
      "(desktop)",
      "(desktop)",
      "(mobile)",
      "(mobile)",
    };
  
  test.assertOutput("data",input,"data_out",output);
}