def test_get_function_set_1(conn): with closing(conn.cursor()) as cursor: reset_db(cursor) conn.commit() with closing(conn.cursor()) as cursor: get_function_set(cursor, 12)
def execute(self): with closing(self.minerva_context.reader_conn.cursor()) as cursor: try: function_set = get_function_set(cursor, self.function_set_id) except NoSuchFunctionSetError: raise JobExecutionError("no function set with id {}".format( self.function_set_id)) transformation = Transformation(function_set, self.dest_timestamp) transformation.execute(self.minerva_context)
def test_get_function_set_2(conn): with closing(conn.cursor()) as cursor: reset_db(cursor) conn.commit() with closing(conn.cursor()) as cursor: entitytype = get_dummy_entitytype(cursor, "dummy_type") src_datasource = get_dummy_datasource(cursor, "dummy-src-1") dst_datasource = get_dummy_datasource(cursor, "dummy-src-2") args = ("test_set", "", [1, 2, 3], [src_datasource.id], entitytype.id, "900", dst_datasource.id, entitytype.id, "900", None, [], None, True) function_set = add_function_set(cursor, *args) function_set = get_function_set(cursor, function_set.id) eq_(function_set.name, 'test_set')