def testSchemaFor_ThroughJythonUDF(self): script = '\n'.join([ "Register 'tests/udfs.py' using jython as udfs;", "data = LOAD '%s' AS (query:CHARARRAY, count:INT);" % self.INPUT_FILE, "queries = FOREACH data GENERATE query, udfs.concat(query,query) AS doublequery;", "STORE queries INTO 'top_3_queries';", ]) proxy = PigProxy(script); schema = proxy.schemaFor('queries') self.assertEqual(schema, '(query: chararray,doublequery: chararray)')
def testSchemaFor_ThroughJythonUDF(self): script = '\n'.join([ "Register 'tests/udfs.py' using jython as udfs;", "data = LOAD '%s' AS (query:CHARARRAY, count:INT);" % self.INPUT_FILE, "queries = FOREACH data GENERATE query, udfs.concat(query,query) AS doublequery;", "STORE queries INTO 'top_3_queries';", ]) proxy = PigProxy(script) schema = proxy.schemaFor('queries') self.assertEqual(schema, '(query: chararray,doublequery: chararray)')