import pytest @pytest.mark.parametrize( 'batch_size,storefronts_number,expected', [ (100, 2, 50), (100, 100, 1), ( 123, 100, 1, ), ] ) def test_service_calculate_number_of_entities( service, batch_size: int, storefronts_number: int, expected: int ): result = service.calculate_number_of_entities(batch_size, storefronts_number) assert result == expected @pytest.mark.parametrize( 'batch_size,storefronts_number,expected', [ (0, 2, ValueError), (100, 0, ValueError), (-100, 1, ValueError), ] ) def test_service_calculate_number_of_entities_raises_error( service, batch_size: int, storefronts_number: int, expected: Exception ): with pytest.raises(expected): service.calculate_number_of_entities(batch_size, storefronts_number)