def create_environment(self):
# Create the maze
self.env = Environment(n=self.ground_dimension,
mine_density=self.mine_density,
visual=self.visual,
end_game_on_mine_hit=self.end_game_on_mine_hit)
self.env.generate_environment()
def generate_traj_set(feat_list):
# Before calling this function, you need to decide what features you care
# about, from a choice of table, coffee, human, origin, and laptop.
pick = [104.2, 151.6, 183.8, 101.8, 224.2, 216.9, 310.8]
place = [210.8, 101.6, 192.0, 114.7, 222.2, 246.1, 322.0]
start = np.array(pick) * (math.pi / 180.0)
goal = np.array(place) * (math.pi / 180.0)
goal_pose = None
T = 20.0
timestep = 0.5
WEIGHTS_DICT = {
"table": [0.0, 0.2, 0.4, 0.6, 0.8, 1.0, 7.0, 8.0],
"coffee": [0.0, 0.2, 0.4, 0.6, 0.8, 1.0],
"laptop": [0.0, 20.0, 21.0, 22.0, 24.0, 26.0, 30.0, 40.0],
"human": [0.0, 20.0, 21.0, 22.0, 24.0, 26.0, 30.0, 40.0],
"efficiency": [1.0]
}
# Openrave parameters for the environment.
model_filename = "jaco_dynamics"
object_centers = {
'HUMAN_CENTER': [-0.6, -0.55, 0.0],
'LAPTOP_CENTER': [-0.7929, -0.1, 0.0]
}
environment = Environment(model_filename, object_centers)
# Planner Setup
max_iter = 50
num_waypts = 5
feat_list = [x.strip() for x in feat_list.split(',')]
num_features = len(feat_list)
planner = TrajoptPlanner(feat_list, max_iter, num_waypts, environment)
# Construct set of weights of interest.
weights_span = [None] * num_features
for feat in range(0, num_features):
weights_span[feat] = WEIGHTS_DICT[feat_list[feat]]
weight_pairs = list(itertools.product(*weights_span))
weight_pairs = [list(i) for i in weight_pairs]
traj_rand = {}
for (w_i, weights) in enumerate(weight_pairs):
traj = planner.replan(start, goal, goal_pose, weights, T, timestep)
Phi = environment.featurize(traj.waypts, feat_list)
# Getting rid of bad, out-of-bounds trajectories
if any(phi < 0.0 for phi in Phi):
continue
traj = traj.waypts.tolist()
if repr(traj) not in traj_rand:
traj_rand[repr(traj)] = weights
here = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../'))
savefile = "/data/traj_sets/traj_set_table_laptop.p"
pickle.dump(traj_rand, open(here + savefile, "wb"))
print "Saved in: ", savefile
print "Used the following weight-combos: ", weight_pairs
def action():
env = Environment().get_environment_info()
capabilities = {'platformName': env.devices[0].platform_name,
'platformVersion': env.devices[0].platform_version,
'deviceName': env.devices[0].device_name,
'appPackage': 'in.togetu.video',
'clearSystemFiles': True,
'appActivity': env.app_activity,
'noReset':True,
'noSign': True
}
host = "http://localhost:4723/wd/hub"
driver = webdriver.Remote(host, capabilities)
yield ElementActions(driver).reset(driver)
driver.quit()
def load_parameters(self):
"""
Loading parameters and setting up variables from the ROS environment.
"""
# ----- General Setup ----- #
self.prefix = rospy.get_param("setup/prefix")
self.start = np.array(rospy.get_param("setup/start"))*(math.pi/180.0)
self.T = rospy.get_param("setup/T")
self.timestep = rospy.get_param("setup/timestep")
self.save_dir = rospy.get_param("setup/save_dir")
# Openrave parameters for the environment.
model_filename = rospy.get_param("setup/model_filename")
object_centers = rospy.get_param("setup/object_centers")
self.environment = Environment(model_filename, object_centers)
# ----- Controller Setup ----- #
# Retrieve controller specific parameters.
controller_type = rospy.get_param("controller/type")
if controller_type == "pid":
# P, I, D gains.
P = rospy.get_param("controller/p_gain") * np.eye(7)
I = rospy.get_param("controller/i_gain") * np.eye(7)
D = rospy.get_param("controller/d_gain") * np.eye(7)
# Stores proximity threshold.
epsilon = rospy.get_param("controller/epsilon")
# Stores maximum COMMANDED joint torques.
MAX_CMD = rospy.get_param("controller/max_cmd") * np.eye(7)
self.controller = PIDController(P, I, D, epsilon, MAX_CMD)
else:
raise Exception('Controller {} not implemented.'.format(controller_type))
# Tell controller to move to start.
self.controller.set_trajectory(Trajectory([self.start], [0.0]))
# Stores current COMMANDED joint torques.
self.cmd = np.eye(7)
# Utilities for recording data.
self.expUtil = experiment_utils.ExperimentUtils(self.save_dir)
def action2():
env = Environment().get_environment_info()
capabilities = {
'platformName': env.devices[0].platform_name,
'platformVersion': env.devices[0].platform_version,
'deviceName': env.devices[0].device_name,
'app': env.apk,
'clearSystemFiles': True,
'appActivity': env.app_activity,
'appPackage': env.app_package,
#'automationName': 'UIAutomator2',
'noSign': True,
'noReset': True,
'systemPort': random.randint(4700, 4900)
}
host = "http://localhost:4724/wd/hub"
driver = webdriver.Remote(host, capabilities)
yield ElementActions(driver).reset(driver)
driver.quit()
from flask import Flask
from flask_restx import Api
from flask_bcrypt import Bcrypt
from flask_jwt_extended import JWTManager
# Configuración Inicial Flask
app = Flask(__name__)
app.config['JWT_SECRET_KEY'] = "12usuario12"
app.config['JWT_ALGORITHM'] = "HS256"
bcrypt = Bcrypt(app)
jwt = JWTManager(app)
api = Api(
app,
title='Usuarios corporativos',
version='2.0',
description='Creando servicio a usuarios corporativos',
)
from utils.environment import Environment
config = Environment().settingsGeneral()
from flask import Flask
from flask_restx import Api
from flask_bcrypt import Bcrypt #JWT
from flask_jwt_extended import JWTManager #JWT
app = Flask(__name__)
app.config['JWT_SECRET_KEY'] = "HolaMensajedePruebas" #JWT
app.config['JWT_ALGORITHM'] = "HS256" #JWT
bcrypt = Bcrypt(app)
jwt = JWTManager(app)
api = Api(
app,
title='Proyecto API CRUD Users',
version='1.0',
description='aplicacion CRUD',
)
from utils.environment import Environment
config = Environment().configGeneral()
def get_vin_box(self):
vin = Environment().get_inited_config().vin_success
box = Environment().get_inited_config().box_success
L.d('车架号:%s 盒子号 %s' % (vin, box))
return [vin, box]
from utils import L
import sys
"""
run all case:
python3 run.py
run one module case:
python3 run.py test/test_home.py
run case with key word:
python3 run.py -k <keyword>
"""
if __name__ == '__main__':
env = Environment()
xml_report_path = env.get_environment_info().xml_report
html_report_path = env.get_environment_info().html_report
# 开始测试
args = [
'-s', '-q', '--allure_features=My stdio1', '--alluredir',
xml_report_path
]
# args = ['-s', '-q'],'--allure_features=My stdio'
self_args = sys.argv[1:]
pytest.main(args)
# 生成html测试报告
cmd = 'allure generate %s -o %s' % (xml_report_path, html_report_path)
#
try:
Shell.invoke(cmd)
class FeatureElicitator():
"""
This class represents a node that moves the Jaco with PID control AND supports receiving human corrections online.
Additionally, it implements a protocol for elicitating novel features from human input.
Subscribes to:
/$prefix$/out/joint_angles - Jaco sensed joint angles
/$prefix$/out/joint_torques - Jaco sensed joint torques
Publishes to:
/$prefix$/in/joint_velocity - Jaco commanded joint velocities
"""
def __init__(self):
# Create ROS node.
rospy.init_node("feature_elicitator")
# Load parameters and set up subscribers/publishers.
self.load_parameters()
self.register_callbacks()
# Start admittance control mode.
ros_utils.start_admittance_mode(self.prefix)
# Publish to ROS at 100hz.
r = rospy.Rate(100)
print "----------------------------------"
print "Moving robot, type Q to quit:"
while not rospy.is_shutdown():
if sys.stdin in select.select([sys.stdin], [], [], 0)[0]:
line = raw_input()
if line == "q" or line == "Q" or line == "quit":
break
self.vel_pub.publish(ros_utils.cmd_to_JointVelocityMsg(self.cmd))
r.sleep()
print "----------------------------------"
ros_utils.stop_admittance_mode(self.prefix)
def load_parameters(self):
"""
Loading parameters and setting up variables from the ROS environment.
"""
# ----- General Setup ----- #
self.prefix = rospy.get_param("setup/prefix")
pick = rospy.get_param("setup/start")
place = rospy.get_param("setup/goal")
self.start = np.array(pick)*(math.pi/180.0)
self.goal = np.array(place)*(math.pi/180.0)
self.goal_pose = None if rospy.get_param("setup/goal_pose") == "None" else rospy.get_param("setup/goal_pose")
self.T = rospy.get_param("setup/T")
self.timestep = rospy.get_param("setup/timestep")
self.save_dir = rospy.get_param("setup/save_dir")
self.INTERACTION_TORQUE_THRESHOLD = rospy.get_param("setup/INTERACTION_TORQUE_THRESHOLD")
self.INTERACTION_TORQUE_EPSILON = rospy.get_param("setup/INTERACTION_TORQUE_EPSILON")
self.CONFIDENCE_THRESHOLD = rospy.get_param("setup/CONFIDENCE_THRESHOLD")
self.N_QUERIES = rospy.get_param("setup/N_QUERIES")
self.nb_layers = rospy.get_param("setup/nb_layers")
self.nb_units = rospy.get_param("setup/nb_units")
# Openrave parameters for the environment.
model_filename = rospy.get_param("setup/model_filename")
object_centers = rospy.get_param("setup/object_centers")
feat_list = rospy.get_param("setup/feat_list")
weights = rospy.get_param("setup/feat_weights")
FEAT_RANGE = rospy.get_param("setup/FEAT_RANGE")
feat_range = [FEAT_RANGE[feat_list[feat]] for feat in range(len(feat_list))]
LF_dict = rospy.get_param("setup/LF_dict")
self.environment = Environment(model_filename, object_centers, feat_list, feat_range, np.array(weights), LF_dict)
# ----- Planner Setup ----- #
# Retrieve the planner specific parameters.
planner_type = rospy.get_param("planner/type")
if planner_type == "trajopt":
max_iter = rospy.get_param("planner/max_iter")
num_waypts = rospy.get_param("planner/num_waypts")
# Initialize planner and compute trajectory to track.
self.planner = TrajoptPlanner(max_iter, num_waypts, self.environment)
else:
raise Exception('Planner {} not implemented.'.format(planner_type))
self.traj = self.planner.replan(self.start, self.goal, self.goal_pose, self.T, self.timestep)
self.traj_plan = self.traj.downsample(self.planner.num_waypts)
# Track if you have reached the start/goal of the path.
self.reached_start = False
self.reached_goal = False
self.feature_learning_mode = False
self.interaction_mode = False
# Save the intermediate target configuration.
self.curr_pos = None
# Track data and keep stored.
self.interaction_data = []
self.interaction_time = []
self.feature_data = []
self.track_data = False
# ----- Controller Setup ----- #
# Retrieve controller specific parameters.
controller_type = rospy.get_param("controller/type")
if controller_type == "pid":
# P, I, D gains.
P = rospy.get_param("controller/p_gain") * np.eye(7)
I = rospy.get_param("controller/i_gain") * np.eye(7)
D = rospy.get_param("controller/d_gain") * np.eye(7)
# Stores proximity threshold.
epsilon = rospy.get_param("controller/epsilon")
# Stores maximum COMMANDED joint torques.
MAX_CMD = rospy.get_param("controller/max_cmd") * np.eye(7)
self.controller = PIDController(P, I, D, epsilon, MAX_CMD)
else:
raise Exception('Controller {} not implemented.'.format(controller_type))
# Planner tells controller what plan to follow.
self.controller.set_trajectory(self.traj)
# Stores current COMMANDED joint torques.
self.cmd = np.eye(7)
# ----- Learner Setup ----- #
constants = {}
constants["step_size"] = rospy.get_param("learner/step_size")
constants["P_beta"] = rospy.get_param("learner/P_beta")
constants["alpha"] = rospy.get_param("learner/alpha")
constants["n"] = rospy.get_param("learner/n")
self.feat_method = rospy.get_param("learner/type")
self.learner = PHRILearner(self.feat_method, self.environment, constants)
def get_performance(self):
performance_dict_1 = dict()
performance_dict_2 = dict()
for mine_density in np.arange(0.01, 0.2, 0.01):
print(mine_density)
performance_dict_1[mine_density] = dict()
performance_dict_2[mine_density] = dict()
final_scores_1 = list()
mines_hit_1 = list()
correct_mines_1 = list()
incorrect_mines_1 = list()
final_scores_2 = list()
mines_hit_2 = list()
correct_mines_2 = list()
incorrect_mines_2 = list()
for _ in range(10):
env = Environment(n=15,
mine_density=mine_density,
end_game_on_mine_hit=False,
visual=False)
env.generate_environment()
agent1 = CSPAgent(env=env, end_game_on_mine_hit=False)
agent1.play()
agent2 = BonusCSPAgent(env=env, end_game_on_mine_hit=False)
agent2.play()
# agent = ProbCSPAgent(env = env,
# end_game_on_mine_hit = False,
# use_probability_agent = self.use_probability_agent,
# prob = 0.3)
# agent.play()
metrics_1 = agent1.get_gameplay_metrics()
final_scores_1.append(metrics_1["final_score"])
mines_hit_1.append(metrics_1["number_of_mines_hit"])
metrics_2 = agent2.get_gameplay_metrics()
final_scores_2.append(metrics_2["final_score"])
mines_hit_2.append(metrics_2["number_of_mines_hit"])
final_score_1 = np.mean(final_scores_1)
num_mines_hit_1 = np.mean(mines_hit_1)
final_score_2 = np.mean(final_scores_2)
num_mines_hit_2 = np.mean(mines_hit_2)
performance_dict_1[mine_density]["final_score"] = final_score_1
performance_dict_1[mine_density][
"number_of_mines_hit"] = num_mines_hit_1
performance_dict_2[mine_density]["final_score"] = final_score_2
performance_dict_2[mine_density][
"number_of_mines_hit"] = num_mines_hit_2
mine_densities_1 = performance_dict_1.keys()
final_scores_1 = [
performance_dict_1[density]["final_score"]
for density in performance_dict_1
]
mines_hit_1 = [
performance_dict_1[density]["number_of_mines_hit"]
for density in performance_dict_1
]
mine_densities_2 = performance_dict_2.keys()
final_scores_2 = [
performance_dict_2[density]["final_score"]
for density in performance_dict_2
]
mines_hit_2 = [
performance_dict_2[density]["number_of_mines_hit"]
for density in performance_dict_2
]
plt.plot(mine_densities_1,
final_scores_1,
marker='o',
label="Normal CSP Agent")
plt.plot(mine_densities_2,
final_scores_2,
marker='o',
label="Bonus CSP Agent")
plt.xlabel("Mine Density")
plt.ylabel("Average Final Score")
plt.savefig('avg_final_score_bonus.png')
plt.legend()
plt.close()
plt.plot(mine_densities_1,
mines_hit_1,
marker='o',
label="Normal CSP Agent")
plt.plot(mine_densities_2,
mines_hit_2,
marker='o',
label="Bonus CSP Agent")
plt.xlabel("Mine Density")
plt.ylabel("Average Density of Mines Hit")
plt.savefig('avg_density_of_mines_hit_bonus.png')
plt.legend()
plt.close()
return performance_dict_1, performance_dict_2
import matplotlib.pyplot as plt
from utils.environment import Environment
print("")
print("#==============================#")
print("# Define Global Variables")
print("#==============================#")
MAP_DIR = "./maps/test/"
MONSTER_DIR = "./monsters/"
WAVE_DIR = "./waves/test/"
print("")
print("#==============================#")
print("# Define the Environment")
print("#==============================#")
environment = Environment(MAP_DIR, MONSTER_DIR, WAVE_DIR)
plt.pause(0.1)
print("")
print("#==============================#")
print("# Walk Monster Path")
print("#==============================#")
print("#")
print("# visualize the walking")
plt.figure(1)
boxProps = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
winText = "You somehow did it ^.^"
looseText = "You lost...as suspected..."
frameRate = 0.01 # [s]
def get_account():
account = Environment().get_inited_config().account_success
pwd = Environment().get_inited_config().password_success
L.d('账号:%s 密码 %s' % (account, pwd))
return [account, pwd]
def load_parameters(self):
"""
Loading parameters and setting up variables from the ROS environment.
"""
# ----- General Setup ----- #
self.prefix = rospy.get_param("setup/prefix")
pick = rospy.get_param("setup/start")
place = rospy.get_param("setup/goal")
self.start = np.array(pick) * (math.pi / 180.0)
self.goal = np.array(place) * (math.pi / 180.0)
self.goal_pose = None if rospy.get_param(
"setup/goal_pose") == "None" else rospy.get_param(
"setup/goal_pose")
self.T = rospy.get_param("setup/T")
self.timestep = rospy.get_param("setup/timestep")
self.save_dir = rospy.get_param("setup/save_dir")
self.feat_list = rospy.get_param("setup/feat_list")
self.weights = rospy.get_param("setup/feat_weights")
self.INTERACTION_TORQUE_THRESHOLD = rospy.get_param(
"setup/INTERACTION_TORQUE_THRESHOLD")
self.INTERACTION_TORQUE_EPSILON = rospy.get_param(
"setup/INTERACTION_TORQUE_EPSILON")
# Openrave parameters for the environment.
model_filename = rospy.get_param("setup/model_filename")
object_centers = rospy.get_param("setup/object_centers")
self.environment = Environment(model_filename, object_centers)
# ----- Planner Setup ----- #
# Retrieve the planner specific parameters.
planner_type = rospy.get_param("planner/type")
if planner_type == "trajopt":
max_iter = rospy.get_param("planner/max_iter")
num_waypts = rospy.get_param("planner/num_waypts")
# Initialize planner and compute trajectory to track.
self.planner = TrajoptPlanner(self.feat_list, max_iter, num_waypts,
self.environment)
else:
raise Exception('Planner {} not implemented.'.format(planner_type))
self.traj = self.planner.replan(self.start, self.goal, self.goal_pose,
self.weights, self.T, self.timestep)
self.traj_plan = self.traj.downsample(self.planner.num_waypts)
# Track if you have reached the start/goal of the path.
self.reached_start = False
self.reached_goal = False
# Save the intermediate target configuration.
self.curr_pos = None
# ----- Controller Setup ----- #
# Retrieve controller specific parameters.
controller_type = rospy.get_param("controller/type")
if controller_type == "pid":
# P, I, D gains.
P = rospy.get_param("controller/p_gain") * np.eye(7)
I = rospy.get_param("controller/i_gain") * np.eye(7)
D = rospy.get_param("controller/d_gain") * np.eye(7)
# Stores proximity threshold.
epsilon = rospy.get_param("controller/epsilon")
# Stores maximum COMMANDED joint torques.
MAX_CMD = rospy.get_param("controller/max_cmd") * np.eye(7)
self.controller = PIDController(P, I, D, epsilon, MAX_CMD)
else:
raise Exception(
'Controller {} not implemented.'.format(controller_type))
# Planner tells controller what plan to follow.
self.controller.set_trajectory(self.traj)
# Stores current COMMANDED joint torques.
self.cmd = np.eye(7)
# ----- Learner Setup ----- #
constants = {}
constants["UPDATE_GAINS"] = rospy.get_param("learner/UPDATE_GAINS")
constants["MAX_WEIGHTS"] = rospy.get_param("learner/MAX_WEIGHTS")
constants["FEAT_RANGE"] = rospy.get_param("learner/FEAT_RANGE")
constants["P_beta"] = rospy.get_param("learner/P_beta")
constants["alpha"] = rospy.get_param("learner/alpha")
constants["n"] = rospy.get_param("learner/n")
self.feat_method = rospy.get_param("learner/type")
self.learner = PHRILearner(self.feat_method, self.feat_list,
self.environment, constants)
# ---- Experimental Utils ---- #
self.expUtil = experiment_utils.ExperimentUtils(self.save_dir)
# Update the list of replanned plans with new trajectory plan.
self.expUtil.update_replanned_trajList(0.0, self.traj_plan.waypts)
# Update the list of replanned waypoints with new waypoints.
self.expUtil.update_replanned_wayptsList(0.0, self.traj.waypts)
def gen_page_py():
GenPages.gen_page_py()
class WatchHandler(PatternMatchingEventHandler):
patterns = ["*.yaml"]
def on_created(self, event):
L.i('监听到文件: yaml 发生了变化')
try:
gen_page_py()
except Exception as e:
L.e('\n!!!!!!!---pages.yaml---!!!!!!\n解析文件 pages.yaml 错误\n'
'请到{}路径下检查修改后重新保存.'.format(self.watch_path))
if __name__ == "__main__":
event_handler = WatchHandler()
full_path = Environment().get_environment_info().pages_yaml
print(full_path)
observer = Observer()
observer.schedule(event_handler, full_path)
observer.start()
try:
while True:
time.sleep(1)
except KeyboardInterrupt:
observer.stop()
observer.join()
def load_parameters(self):
"""
Loading parameters and setting up variables from the ROS environment.
"""
# ----- General Setup ----- #
self.prefix = rospy.get_param("setup/prefix")
pick = rospy.get_param("setup/start")
place = rospy.get_param("setup/goal")
self.start = np.array(pick) * (math.pi / 180.0)
self.goal = np.array(place) * (math.pi / 180.0)
self.goal_pose = None if rospy.get_param(
"setup/goal_pose") == "None" else rospy.get_param(
"setup/goal_pose")
self.T = rospy.get_param("setup/T")
self.timestep = rospy.get_param("setup/timestep")
self.feat_list = rospy.get_param("setup/feat_list")
self.weights = rospy.get_param("setup/feat_weights")
# Openrave parameters for the environment.
model_filename = rospy.get_param("setup/model_filename")
object_centers = rospy.get_param("setup/object_centers")
self.environment = Environment(model_filename, object_centers)
# ----- Planner Setup ----- #
# Retrieve the planner specific parameters.
planner_type = rospy.get_param("planner/type")
if planner_type == "trajopt":
max_iter = rospy.get_param("planner/max_iter")
num_waypts = rospy.get_param("planner/num_waypts")
# Initialize planner and compute trajectory to track.
self.planner = TrajoptPlanner(self.feat_list, max_iter, num_waypts,
self.environment)
else:
raise Exception('Planner {} not implemented.'.format(planner_type))
self.traj = self.planner.replan(self.start, self.goal, self.goal_pose,
self.weights, self.T, self.timestep)
# Save the intermediate target configuration.
self.curr_pos = None
# ----- Controller Setup ----- #
# Retrieve controller specific parameters.
controller_type = rospy.get_param("controller/type")
if controller_type == "pid":
# P, I, D gains.
P = rospy.get_param("controller/p_gain") * np.eye(7)
I = rospy.get_param("controller/i_gain") * np.eye(7)
D = rospy.get_param("controller/d_gain") * np.eye(7)
# Stores proximity threshold.
epsilon = rospy.get_param("controller/epsilon")
# Stores maximum COMMANDED joint torques.
MAX_CMD = rospy.get_param("controller/max_cmd") * np.eye(7)
self.controller = PIDController(P, I, D, epsilon, MAX_CMD)
else:
raise Exception(
'Controller {} not implemented.'.format(controller_type))
# Planner tells controller what plan to follow.
self.controller.set_trajectory(self.traj)
# Stores current COMMANDED joint torques.
self.cmd = np.eye(7)
def _run_thinning_astar(self, root, dest):
current_maze_copy = self.environment.maze.copy()
simpler_maze = self._create_simpler_maze(maze=current_maze_copy)
simpler_env = Environment(n=self.environment.n, p=self.environment.p)
simpler_env.generate_maze(new_maze=simpler_maze)
# Root is at a distance of 0 from itself
root.distance_from_source = 0
self.fringe = Q.PriorityQueue()
self.fringe.put(root)
self.visited.append(root)
while self.fringe.queue:
# Keep track of maximum fringe length
fringe_length = len(self.fringe.queue)
if fringe_length >= self.max_fringe_length:
self.max_fringe_length = fringe_length
temp_path = []
node = self.fringe.get()
if node not in self.visited:
self.visited.append(node)
node_children = node.get_children(node=node,
algorithm=self.algorithm)
for child in node_children:
if child is None or child in self.visited:
continue
if child not in self.fringe.queue:
child.parent = node
temp_simpler_maze = simpler_maze.copy()
temp_simpler_maze[child.row, child.column] = 4
simpler_env.modify_environment(new_maze=temp_simpler_maze)
child.distance_from_dest = self._calculate_thinning_heuristic(
simpler_maze_env=simpler_env,
node_row=child.row,
node_column=child.column)
child.distance_from_source = node.distance_from_source + 1
self.fringe.put(child)
else:
if child.get_heuristic(
) >= node.distance_from_source + child.distance_from_dest:
child.parent = node
child.distance_from_source = node.distance_from_source + 1
# Get the path through which you reach this node from the root node
flag = True
temp_node = node
while (flag):
temp_path.append(temp_node)
temp_node = temp_node.parent
if temp_node is None:
flag = False
temp_path_copy = temp_path.copy()
# Update the color of the path which we found above by popping the root first and the subsequent nodes.
while (len(temp_path) != 0):
temp_node = temp_path.pop()
# Visualisation Parameter added
if self.visual == True:
self.environment.update_color_of_cell(
temp_node.row, temp_node.column)
self.environment.render_maze(title=self.title)
# if you reach the destination, then break
if (node == dest):
break
# We reset the entire path again to render a new path in the next iteration.
while (len(temp_path_copy) != 0):
temp_node = temp_path_copy.pop(0)
# Visualisation Parameter added
if self.visual == True:
self.environment.reset_color_of_cell(
temp_node.row, temp_node.column)
self.environment.render_maze(title=self.title)
def plot_performance_metrics(self): algo_performance = dict() env = Environment(n = 20, p = self.probability_of_obstacles, fire = self.fire) # env.generate_maze() # env.create_graph_from_maze() for n in np.arange(10, 100, 10): print(n) for algo in ["astar"]: for heuristic in ["euclid", "manhattan"]: print(algo, heuristic) if (algo + "_" + heuristic) not in algo_performance: algo_performance[(algo + "_" + heuristic)] = dict() if algo not in algo_performance: algo_performance[algo] = dict() env.algorithm = algo env.create_graph_from_maze() path_finder = PathFinderAlgorithm(environment = env, algorithm = algo, visual = self.visual, heuristic = self.heuristic) path_finder.run_path_finder_algorithm() performances = path_finder.performance_dict if "path_length" not in algo_performance[(algo + "_" + heuristic)]: algo_performance[(algo + "_" + heuristic)]["path_length"] = [] if "maximum_fringe_size" not in algo_performance[(algo + "_" + heuristic)]: algo_performance[(algo + "_" + heuristic)]["maximum_fringe_size"] = [] if "number_of_nodes_expanded" not in algo_performance[(algo + "_" + heuristic)]: algo_performance[(algo + "_" + heuristic)]["number_of_nodes_expanded"] = [] algo_performance[(algo + "_" + heuristic)]['path_length'].append(performances['path_length']) algo_performance[(algo + "_" + heuristic)]['maximum_fringe_size'].append(performances['maximum_fringe_size']) algo_performance[(algo + "_" + heuristic)]['number_of_nodes_expanded'].append(performances['number_of_nodes_expanded']) for algo in ["astar", "thin_astar"]: for heuristic in ["euclid", "manhattan"]: print(algo, heuristic) if (algo + "_" + heuristic) not in algo_performance: algo_performance[(algo + "_" + heuristic)] = dict() env.algorithm = algo path_finder = PathFinderAlgorithm(environment = env, algorithm = algo, q = q, visual = self.visual, heuristic = heuristic) path_finder.run_path_finder_algorithm() performances = path_finder.performance_dict if "path_length" not in algo_performance[(algo + "_" + heuristic)]: algo_performance[(algo + "_" + heuristic)]["path_length"] = [] if "maximum_fringe_size" not in algo_performance[(algo + "_" + heuristic)]: algo_performance[(algo + "_" + heuristic)]["maximum_fringe_size"] = [] if "number_of_nodes_expanded" not in algo_performance[(algo + "_" + heuristic)]: algo_performance[(algo + "_" + heuristic)]["number_of_nodes_expanded"] = [] algo_performance[(algo + "_" + heuristic)]['path_length'].append(performances['path_length']) algo_performance[(algo + "_" + heuristic)]['maximum_fringe_size'].append(performances['maximum_fringe_size']) algo_performance[(algo + "_" + heuristic)]['number_of_nodes_expanded'].append(performances['number_of_nodes_expanded']) return algo_performance
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Created by zhouyuyan on 2017/5/3 14:14
from time import sleep
from Common.Element import *
from Common.action import ElementActions
from utils.environment import Environment
env = Environment().get_environment_info()
# """"登录页面测试步骤"""
#
def goto_login(self):
global action
action = ElementActions(driver=self.driver)
sleep(4)
my = get_name(self, "我的")
check_my = lambda action: action.is_element_displayed(my)
if check_my:
my.click()
sleep(2)
try:
lgbs = get_name(self, '登录')
check_lgbs = lambda action: action.is_element_displayed(lgbs)
if check_lgbs:
lgbs.click()
sleep(2)
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Librerias importadas
from utils.environment import Environment
import unittest
import json
# Modulos propios
from application import app
from utils.userGenerator import UserGenerator
generateUser = UserGenerator().generateUser
BASE_URL = "http://localhost:8080"
SIGNUP_URL = f"{BASE_URL}/home/signup"
LOGIN_URL = f"{BASE_URL}/home/login"
CHAT_URL = f"{BASE_URL}/chat"
FRIENDS_URL = f"{BASE_URL}/dashboard/friends"
PROFILE_URL = f"{BASE_URL}/dashboard"
MONGO = Environment().settingsDB()