def test_loading_and_storing_model_in_kvs(self):
path = os.path.join(test.SCHEMA_DIR, TEST_FILE)
vulnerability.load_vulnerability_model(1234, path)
model = vulnerability.load_vuln_model_from_kvs(1234)
self.assertEqual(NO_OF_CURVES_IN_TEST_FILE, len(model))
expected_curve = shapes.Curve(
[
(5.0, (0.00, 0.3)),
(5.5, (0.00, 0.3)),
(6.0, (0.00, 0.3)),
(6.5, (0.00, 0.3)),
(7.0, (0.00, 0.3)),
(7.5, (0.01, 0.3)),
(8.0, (0.06, 0.3)),
(8.5, (0.18, 0.3)),
(9.0, (0.36, 0.3)),
(9.5, (0.36, 0.3)),
(10.0, (0.36, 0.3)),
]
)
self.assertEqual(expected_curve, model["PK"])
self.assertEqual(expected_curve, model["IR"])
def setUp(self):
# get random ID as job_id
self.job_id = str(uuid.uuid4())
# load vulnerability and write to memcache
vulnerability.load_vulnerability_model(self.job_id,
os.path.join(test.DATA_DIR, TEST_FILE))
self.vulnerability_curves = \
vulnerability.load_vuln_curves_from_kvs(self.job_id)
def store_vulnerability_model(self):
""" load vulnerability and write to kvs """
path = os.path.join(
self.calc_proxy.base_path,
self.calc_proxy.params["VULNERABILITY"])
vulnerability.load_vulnerability_model(self.calc_proxy.job_id, path)
if self.is_benefit_cost_ratio_mode():
path = os.path.join(
self.calc_proxy.base_path,
self.calc_proxy.params["VULNERABILITY_RETROFITTED"])
vulnerability.load_vulnerability_model(
self.calc_proxy.job_id, path, retrofitted=True)
def store_vulnerability_model(self):
""" load vulnerability and write to kvs """
path = os.path.join(self.job_ctxt.base_path,
self.job_ctxt.params["VULNERABILITY"])
vulnerability.load_vulnerability_model(self.job_ctxt.job_id, path)
if self.is_benefit_cost_ratio_mode():
path = os.path.join(
self.job_ctxt.base_path,
self.job_ctxt.params["VULNERABILITY_RETROFITTED"])
vulnerability.load_vulnerability_model(self.job_ctxt.job_id,
path,
retrofitted=True)
def test_loading_and_storing_model_in_kvs(self):
path = os.path.join(test.SCHEMA_DIR, TEST_FILE)
vulnerability.load_vulnerability_model(1234, path)
model = vulnerability.load_vuln_model_from_kvs(1234)
self.assertEqual(NO_OF_CURVES_IN_TEST_FILE, len(model))
expected_curve = shapes.Curve([(5.0, (0.00, 0.3)), (5.5, (0.00, 0.3)),
(6.0, (0.00, 0.3)), (6.5, (0.00, 0.3)),
(7.0, (0.00, 0.3)), (7.5, (0.01, 0.3)),
(8.0, (0.06, 0.3)), (8.5, (0.18, 0.3)),
(9.0, (0.36, 0.3)), (9.5, (0.36, 0.3)),
(10.0, (0.36, 0.3))])
self.assertEqual(expected_curve, model["PK"])
self.assertEqual(expected_curve, model["IR"])
def test_loading_and_storing_model_in_kvs(self):
path = os.path.join(helpers.SCHEMA_DIR, TEST_FILE)
vulnerability.load_vulnerability_model(1234, path)
model = vulnerability.load_vuln_model_from_kvs(1234)
self.assertEqual(NO_OF_CURVES_IN_TEST_FILE, len(model))
exp_imls = [5.0, 5.5, 6.0, 6.5, 7.0, 7.5, 8.0, 8.5, 9.0, 9.5, 10.0]
exp_loss_ratios = [0.00, 0.00, 0.00, 0.00, 0.00, 0.01, 0.06, 0.18,
0.36, 0.36, 0.36]
exp_covs = [0.3] * 11
expected_curve = shapes.VulnerabilityFunction(exp_imls,
exp_loss_ratios, exp_covs)
self.assertEqual(expected_curve, model["PK"])
self.assertEqual(expected_curve, model["IR"])
def test_loading_and_storing_model_in_kvs(self):
path = os.path.join(helpers.SCHEMA_DIR, TEST_FILE)
vulnerability.load_vulnerability_model(1234, path)
model = vulnerability.load_vuln_model_from_kvs(1234)
self.assertEqual(NO_OF_CURVES_IN_TEST_FILE, len(model))
exp_imls = [5.0, 5.5, 6.0, 6.5, 7.0, 7.5, 8.0, 8.5, 9.0, 9.5, 10.0]
exp_loss_ratios = [
0.00, 0.00, 0.00, 0.00, 0.00, 0.01, 0.06, 0.18, 0.36, 0.36, 0.36
]
exp_covs = [0.3] * 11
expected_curve = shapes.VulnerabilityFunction(exp_imls,
exp_loss_ratios,
exp_covs)
self.assertEqual(expected_curve, model["PK"])
self.assertEqual(expected_curve, model["IR"])
def store_vulnerability_model(self):
""" load vulnerability and write to kvs """
vulnerability.load_vulnerability_model(self.job_id,
os.path.join(self.base_path, self.params["VULNERABILITY"]))
def store_vulnerability_model(self):
""" load vulnerability and write to kvs """
vulnerability.load_vulnerability_model(self.id,
"%s/%s" % (self.base_path, self.params["VULNERABILITY"]))
