def make_sink(self, table_to_schema, table_name, schema, write_disposition): # pylint: disable=unused-argument self.assertEqual(table_to_schema[table_name], schema) return beam_testutil.FakeSink(table_name)
def make_sink(table_name, schema, write_disposition): # pylint: disable=unused-argument return beam_testutil.FakeSink(table_name)
def make_sink(self, table_to_schema, table_name, schema, write_disposition): # pylint: disable=unused-argument expected_schema = set(table_to_schema[table_name].split(', ')) result_schema = set(schema.split(', ')) self.assertEqual(expected_schema, result_schema) return beam_testutil.FakeSink(table_name)