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')