""" Archivo para Test de AI """ from collections import namedtuple from TEST.tools.request import request # Servidor a ejecutar ########### SERVER = "http://127.0.0.1:1115" # Parámetros necesarios ########### USER = 'test' FtpConnection = namedtuple('ftp_connection', ['protocol', 'ip', 'port', 'user', 'password']) LocalFtp = FtpConnection('FTP', '192.168.1.43', '30018', 'marketing', 'marketing') # Lista con el tipo de servicio, las urls y el data ########### TESTS = [ # TODO: ESTOS TEST SOLO FUNCIONAN SI SE CAMBIA EL PARSER DE # PARSER_NONBLOCKING A PARSER_NONLOCAL en detection_by_entity. # Y SE AÑADE AL PARSER_NONE EN EL # parser_in SE MODIFICA UNA LÍNEA TAL QUE: # and not issubclass(sign.parameters[sign_parameter].annotation.__class__ ########### detection ######## ["post", "/AI/skyNet/detection", { 'user': USER, 'aoi': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "KML", 'type': 'VECTOR', "source": '/KML/AOI_kml.kml', 'layer_name': "AOI_kml"}, 'orthophoto': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "GeoTIFF", 'type': 'RASTER', 'layer_name': "trigueros", "source": '/COG/trigueros.tif'}, 'lidar': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "GeoTIFF", "type": "RASTER", 'layer_name': "", "source": '/geotiff'}, 'ms': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "GeoTIFF", "type": "RASTER", 'layer_name': "", "source": '/MS'}, 'output_layer': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "SHAPE", 'type': 'VECTOR', 'layer_name': "result", "source": '/SHP/result_shp.shp'}, "confident": 0, 'indexes': ['calzada_no_revestida', 'calzada_revestida', 'calzada_revestida_urbana'], 'is_optimized': 'false', 'model': 'UNetWays'} ], ################ POST detection_by_entity ###################### ["post", "/AI/skyNet/detection_by_entity", { 'user': USER, 'aoi': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "KML", 'type': 'VECTOR', "source": '/sonia/KML/breeze.kml', 'layer_name': ""}, 'orthophoto': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "GeoTIFF", 'type': 'RASTER', 'layer_name': "", "source": '/sonia/GEOTIFF/castillo_google.tif'}, 'lidar': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "GeoTIFF", "type": "RASTER", 'layer_name': "", "source": '/sonia/ia'}, 'ms': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "GeoTIFF", "type": "RASTER", 'layer_name': "", "source": '/sonia/ia'}, 'output_layer': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "SHAPE", 'type': 'VECTOR', 'layer_name': "result_model_UNetWays_22", "source": '/sonia/shape/result_model_UNetWays_22.shp'}, "confident": 0.2, 'entities': ['calzada_no_revestida', 'calzada_revestida', 'calzada_revestida_urbana'], 'is_optimized': 'false' }], ############### GET detection_by_entity ###################### ["get", "/AI/skyNet/detection_by_entity", { 'user': USER, }], ################# focused_training ###################### ["post", "/AI/skyNet/focused_training", { 'user': USER, 'aois': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "layer_name": "castillo_locubin", 'type': 'VECTOR', "driver_type": "KML", "source": "/KML/castillo_locubin.kml"}, 'orthophoto': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "layer_name": "", 'type': 'RASTER', "driver_type": "GeoTIFF", "source": "/COG/castillo_locubin.tif"}, 'entities': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "layer_name": "entities_cl", 'type': 'VECTOR', "driver_type": "KML", "source": "/KML/entities_cl.kml"}, 'output_layer': { 'protocol': LocalFtp.protocol, "ip": LocalFtp.ip, "port": LocalFtp.port, "user": LocalFtp.user, "password": LocalFtp.password, "driver_type": "SHAPE", 'type': 'VECTOR', 'layer_name': "result_focused_training", "source": '/SHP/result_focused_training.shp'}, 'name': 'focused_train_1', 'zoom_levels': [20, 40], 'image_max_size': 256, 'image_min_size': 256, 'train_percentage': 80, 'empty_prob': 0.5, 'epochs':5, 'validation_steps': -1, 'transfer_learning': True, 'unfrozen_layers': 25, 'is_annealed': True, 'lr': 0.0001, 'lr_schedule_type': 'linear', 'lr_constant_epoch': 5, 'data_aug': True, 'data_aug_args': None, 'regularizer': None, 'confidence': 0.2 }], ################ GET focused_training ###################### ["get", "/AI/skyNet/focused_training", { 'user': USER, }], ##### axis_truck_ai_task ########## ["post", "/GMS/vectorial/axis_road_ai_task", { 'user': USER, 'input_layer': { 'protocol': 'PostgreSQL', 'type': 'VECTOR', 'driver_type': 'PostgreSQL', 'schema': 'ZxasCabW7iWwiBeL8', "ip": '192.168.1.43', "domain": '', "port": '30030', "user": 'srm_postgres_1', "database_name": 'imetadata', "password": 'pg_produccion#1', "table_view_name": 'calzada_revestida_urbana__raw_20220331T153229'}, 'output_layer': { 'protocol': 'PostgreSQL', 'type': 'VECTOR', 'driver_type': 'PostgreSQL', 'schema': 'ZxasCabW7iWwiBeL8', "ip": '192.168.1.43', "domain": '', "port": '30030', "user": 'srm_postgres_1', "database_name": 'imetadata', "password": 'pg_produccion#1', "table_view_name": 'simplify_areas_pr'}, 'output_layer_1': { 'protocol': 'PostgreSQL', 'type': 'VECTOR', 'driver_type': 'PostgreSQL', 'schema': 'ZxasCabW7iWwiBeL8', "ip": '192.168.1.43', "domain": '', "port": '30030', "user": 'srm_postgres_1', "database_name": 'imetadata', "password": 'pg_produccion#1', "table_view_name": 'axis_pr', 'autorefresh': True} }], [] ] if __name__ == '__main__': for test in TESTS[:-1]: try: request(user=USER, type_request=test[0], server=SERVER, service=test[1], data=test[2], time_sleep=5) except Exception as error: print("ERROR:" + str(error))