def test_insert_risk(self):
dycast_parameters = test_helper_functions.get_dycast_parameters()
risk_service = risk_service_module.RiskService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
gridpoints = geography_service.generate_grid(dycast_parameters)
point = geography_service.get_shape_from_sqlalch_element(gridpoints[0])
risk = Risk(risk_date=datetime.date(int(2016), int(3), int(25)),
number_of_cases=5,
lat=point.x,
long=point.y,
close_pairs=3,
close_space=2,
close_time=1,
cumulative_probability=0.032)
session.query(Risk.risk_date).filter(Risk.risk_date == risk.risk_date,
Risk.lat == risk.lat,
Risk.long == risk.long) \
.delete()
risk_service.insert_risk(session, risk)
session.commit()
session.query(Risk.risk_date).filter(Risk.risk_date == risk.risk_date,
Risk.lat == risk.lat,
Risk.long == risk.long) \
.one()
def test_close_time_only(self):
dycast_parameters = test_helper_functions.get_dycast_parameters()
risk_service = risk_service_module.RiskService(dycast_parameters)
comparative_test_service = comparative_test_service_module.ComparativeTestService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
riskdate = datetime.date(int(2016), int(3), int(25))
gridpoints = geography_service.generate_grid(dycast_parameters)
clusters_per_point_query = risk_service.get_clusters_per_point_query(session, gridpoints, riskdate)
clusters_per_point = risk_service.get_clusters_per_point_from_query(clusters_per_point_query)
risk_service.enrich_clusters_per_point_with_close_space_and_time(clusters_per_point)
# Compare to old query
daily_cases_query = comparative_test_service.get_daily_cases_query(session,
riskdate)
for point in gridpoints:
cases_in_cluster_query = comparative_test_service.get_cases_in_cluster_query(daily_cases_query,
point)
count_old = comparative_test_service.get_close_time_only(cases_in_cluster_query)
for cluster in clusters_per_point:
if cluster.point.equals(geography_service.get_shape_from_sqlalch_element(point)):
self.assertEquals(cluster.close_in_time, count_old)
def test_get_clusters_per_point_query(self):
dycast_parameters = test_helper_functions.get_dycast_parameters(large_dataset=True)
risk_service = risk_service_module.RiskService(dycast_parameters)
comparative_test_service = comparative_test_service_module.ComparativeTestService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
riskdate = datetime.date(int(2016), int(3), int(25))
gridpoints = geography_service.generate_grid(dycast_parameters)
clusters_per_point_query = risk_service.get_clusters_per_point_query(session, gridpoints, riskdate)
clusters_per_point = risk_service.get_clusters_per_point_from_query(clusters_per_point_query)
daily_cases_query = comparative_test_service.get_daily_cases_query(session, riskdate)
for cluster in clusters_per_point:
point_wkt_element = geography_service.get_wktelement_from_wkt(cluster.point.wkt)
cases_in_cluster_query = comparative_test_service.get_cases_in_cluster_query(daily_cases_query,
point_wkt_element)
vector_count_new = cluster.get_case_count()
vector_count_old = database_service.get_count_for_query(cases_in_cluster_query)
self.assertEqual(vector_count_new, vector_count_old)
def test_generate_risk(self):
dycast_parameters = test_helper_functions.get_dycast_parameters(large_dataset=False)
risk_service = risk_service_module.RiskService(dycast_parameters)
import_service = import_service_module.ImportService()
import_service.load_case_files(dycast_parameters)
risk_service.generate_risk()
risk_count = test_helper_functions.get_count_from_table("risk")
self.assertGreaterEqual(risk_count, 6)
def test_get_daily_cases_query_old(self):
dycast_parameters = test_helper_functions.get_dycast_parameters()
comparative_test_service = comparative_test_service_module.ComparativeTestService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
riskdate = datetime.date(int(2016), int(3), int(25))
daily_cases_query = comparative_test_service.get_daily_cases_query(session, riskdate)
count = database_service.get_count_for_query(daily_cases_query)
self.assertGreater(count, 0)
def test_get_cumulative_probability(self):
dycast_parameters = test_helper_functions.get_dycast_parameters()
risk_service = risk_service_module.RiskService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
cluster = Cluster()
cluster.case_count = 10
cluster.close_space_and_time = 3
cluster.close_in_space = 5
cluster.close_in_time = 27
risk_service.get_cumulative_probability_for_cluster(session, cluster)
self.assertGreater(cluster.cumulative_probability, 0)
def test_get_exact_match_distribution_margin(self):
dycast_parameters = test_helper_functions.get_dycast_parameters()
risk_service = risk_service_module.RiskService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
cluster = Cluster()
cluster.case_count = 2,
cluster.close_space_and_time = 1,
cluster.close_in_space = 1,
cluster.close_in_time = 1,
cumulative_probability = risk_service.get_exact_match_cumulative_probability(session, cluster)
self.assertGreater(cumulative_probability, 0)
def test_get_nearest_close_in_time_distribution_margin_query(self):
dycast_parameters = test_helper_functions.get_dycast_parameters()
risk_service = risk_service_module.RiskService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
cluster = Cluster()
cluster.case_count = 30
cluster.close_space_and_time = 1
cluster.close_in_space = 2
cluster.close_in_time = 10
nearest_close_in_time_query = risk_service.get_nearest_close_in_time_distribution_margin_query(session, cluster)
result = session.query(nearest_close_in_time_query).first()
self.assertIsNotNone(result)
self.assertGreaterEqual(result, 0)
def test_get_cumulative_probability_by_nearest_close_in_time_and_space(self):
dycast_parameters = test_helper_functions.get_dycast_parameters()
risk_service = risk_service_module.RiskService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
cluster = Cluster()
cluster.case_count = 30
cluster.close_space_and_time = 1
cluster.close_in_space = 2
cluster.close_in_time = 10
query_result = risk_service.get_cumulative_probability_by_nearest_close_in_time_and_space(session,
cluster)
cumulative_probability = query_result.cumulative_probability
self.assertIsNotNone(cumulative_probability)
self.assertGreater(cumulative_probability, 0)
def test_get_cases_in_cluster_query_old(self):
dycast_parameters = test_helper_functions.get_dycast_parameters()
comparative_test_service = comparative_test_service_module.ComparativeTestService(dycast_parameters)
session = database_service.get_sqlalchemy_session()
riskdate = datetime.date(int(2016), int(3), int(25))
gridpoints = geography_service.generate_grid(dycast_parameters)
point = gridpoints[0]
daily_cases_query = comparative_test_service.get_daily_cases_query(session, riskdate)
cases_in_cluster_query = comparative_test_service.get_cases_in_cluster_query(daily_cases_query, point)
vector_count = database_service.get_count_for_query(cases_in_cluster_query)
self.assertGreater(vector_count, 0)