def test_calculate_linear_or_presize_linear_length_test_results_sanity(
self):
"""
test if the numerator is greater then denominator (if the subgroup is greater then containing group)
"""
scene_tool_box = PngcnSceneKpis(self.ProjectConnector_mock,
self.common_mock, 16588190,
self.data_provider_mock)
data = [{
'scene_fk': 101,
'manufacturer_fk': 2,
'product_fk': 252,
'width_mm': 0.84,
'width_mm_advance': 1.23
}, {
'scene_fk': 121,
'manufacturer_fk': 4,
'product_fk': 132,
'width_mm': 0.80,
'width_mm_advance': 0.99
}, {
'scene_fk': 201,
'manufacturer_fk': 4,
'product_fk': 152,
'width_mm': 0.28,
'width_mm_advance': 0.75
}, {
'scene_fk': 151,
'manufacturer_fk': 5,
'product_fk': 172,
'width_mm': 0.95,
'width_mm_advance': 0.15
}]
scene_tool_box.get_filterd_matches = MagicMock(
return_value=pd.DataFrame(data))
scene_tool_box.png_manufacturer_fk = 4
scene_tool_box.common.write_to_db_result = MagicMock()
width = random.choice(['width_mm', 'width_mm_advance'])
scene_tool_box.calculate_linear_or_presize_linear_length(width)
kpi_results = scene_tool_box.common.write_to_db_result.mock_calls[0][2]
if kpi_results:
numerator = kpi_results['numerator_result']
denominator = kpi_results['denominator_result']
self.assertGreaterEqual(
denominator, numerator,
'the numerator cant be greater then denominator')
def test_calculate_linear_or_presize_linear_length_test_results_length(
self):
"""
test that we write 8 fields to DB
"""
scene_tool_box = PngcnSceneKpis(self.ProjectConnector_mock,
self.common_mock, 16588190,
self.data_provider_mock)
data = [{
'scene_fk': 101,
'manufacturer_fk': 2,
'product_fk': 252,
'width_mm': 0.84,
'width_mm_advance': 1.23
}, {
'scene_fk': 121,
'manufacturer_fk': 4,
'product_fk': 132,
'width_mm': 0.80,
'width_mm_advance': 0.99
}, {
'scene_fk': 201,
'manufacturer_fk': 4,
'product_fk': 152,
'width_mm': 0.28,
'width_mm_advance': 0.75
}, {
'scene_fk': 151,
'manufacturer_fk': 5,
'product_fk': 172,
'width_mm': 0.95,
'width_mm_advance': 0.15
}]
scene_tool_box.get_filterd_matches = MagicMock(
return_value=pd.DataFrame(data))
scene_tool_box.png_manufacturer_fk = 4
scene_tool_box.common.write_to_db_result = MagicMock()
width = random.choice(['width_mm', 'width_mm_advance'])
scene_tool_box.calculate_linear_or_presize_linear_length(width)
kpi_results = scene_tool_box.common.write_to_db_result.mock_calls[0][2]
if kpi_results:
self.assertEqual(len(kpi_results), 8,
'expects to write 8 parameters to db')
