from Trax.Cloud.Services.Connector.Logger import LoggerInitializer
from Projects.NESTLEUS.Calculations import Calculations
from Projects.NESTLEUS.Utils import Const
import pandas as pd
def checkout(sessions):
for session in sessions:
data_provider.load_session_data(session)
output = Output()
Calculations(data_provider, output).run_project_calculations()
if __name__ == '__main__':
LoggerInitializer.init('nestleus calculations')
Config.init()
project_name = 'nestleus'
data_provider = KEngineDataProvider(project_name)
checkout([
#'0d71ea3c-6430-488e-88e1-dd83ef64d2c9'
# '177c0014-46e6-41cc-a35d-2e2c462a2537',
# '50730f02-8ac1-4918-bc48-acf6fc02dfb6',
# '6a54ba4b-c2da-4939-bb70-79101f4016e1',
# 'd318462b-324d-4426-abcf-21d191c6170d',
# '6a54ba4b-c2da-4939-bb70-79101f4016e1'
])
test_sessions = pd.read_excel(Const.TEST_SESSIONS_PATH)
i = 0.0
def __init__(self):
env = Config.get_environment()
self._base_file_path = os.path.join(local_repository_path(), env,
'sftp')
create_folder(self._base_file_path)
def create_graph_image(scene_id, graph):
filtered_figure = GraphPlot.plot_networkx_graph(
graph,
overlay_image=True,
scene_id=scene_id,
project_name='diageous-sand2')
filtered_figure.update_layout(autosize=False,
width=1000,
height=800,
title=str(scene_id))
iplot(filtered_figure)
if __name__ == '__main__':
from KPIUtils_v2.DB.CommonV2 import Common
from Trax.Utils.Conf.Configuration import Config
from Trax.Algo.Calculations.Core.DataProvider import KEngineDataProvider
from Trax.Cloud.Services.Connector.Logger import LoggerInitializer
LoggerInitializer.init('KEngine')
Config.init('KEngine')
test_data_provider = KEngineDataProvider('diageous')
sessions = ['2087F0BA-E12A-458A-83D0-0713E9DF1EBA']
for session in sessions:
print(session)
test_data_provider.load_session_data(session)
test_common = Common(test_data_provider)
case_counter_calculator = CaseCountCalculator(test_data_provider,
test_common)
case_counter_calculator.main_case_count_calculations()
test_common.commit_results_data()
res = self.merged_kpi_results.loc[self.merged_kpi_results['client_name'] == row['client_name']]
if res.empty:
pass
else:
res = res.groupby(['result'])
if len(res.count().index) < 2:
print '## there is only 2 result type for kpi ' + row['client_name']
print res.count().get_values()
def run_all_tests(self):
self.test_invalid_precent_results()
self.test_uncalculated_kpi()
self.test_result_is_zero()
self.test_results_in_expected_range()
self.test_one_result_in_all_sessions()
#TODO
# 1.pull data from prod base on dates
# 2.plot histogram
if __name__ == "__main__":
Config.init(app_name='ttt', default_env='prod',
config_file='~/theGarage/Trax/Apps/Services/KEngine/k-engine-prod.config')
qa_tool = qa('diageous', start_date='2018-05-01', end_date='2018-08-01')
