mirror of
https://github.com/capnproto/pycapnp.git
synced 2025-03-04 08:24:43 +01:00
6e7fffd7de
- Fixing flake8 configuration to agree with black - Adding black validation check to github actions
110 lines
2.9 KiB
Python
Executable file
110 lines
2.9 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import carsales_pb2
|
|
from common import rand_int, rand_double, rand_bool, from_bytes_helper
|
|
from random import choice
|
|
|
|
MAKES = ["Toyota", "GM", "Ford", "Honda", "Tesla"]
|
|
MODELS = ["Camry", "Prius", "Volt", "Accord", "Leaf", "Model S"]
|
|
COLORS = [
|
|
"black",
|
|
"white",
|
|
"red",
|
|
"green",
|
|
"blue",
|
|
"cyan",
|
|
"magenta",
|
|
"yellow",
|
|
"silver",
|
|
]
|
|
|
|
|
|
def random_car(car):
|
|
car.make = choice(MAKES)
|
|
car.model = choice(MODELS)
|
|
car.color = rand_int(len(COLORS))
|
|
|
|
car.seats = 2 + rand_int(6)
|
|
car.doors = 2 + rand_int(3)
|
|
|
|
for _ in range(4):
|
|
wheel = car.wheel.add()
|
|
wheel.diameter = 25 + rand_int(15)
|
|
wheel.air_pressure = 30 + rand_double(20)
|
|
wheel.snow_tires = rand_int(16) == 0
|
|
|
|
car.length = 170 + rand_int(150)
|
|
car.width = 48 + rand_int(36)
|
|
car.height = 54 + rand_int(48)
|
|
car.weight = car.length * car.width * car.height // 200
|
|
|
|
engine = car.engine
|
|
engine.horsepower = 100 * rand_int(400)
|
|
engine.cylinders = 4 + 2 * rand_int(3)
|
|
engine.cc = 800 + rand_int(10000)
|
|
engine.uses_gas = True
|
|
engine.uses_electric = rand_bool()
|
|
|
|
car.fuel_capacity = 10.0 + rand_double(30.0)
|
|
car.fuel_level = rand_double(car.fuel_capacity)
|
|
car.has_power_windows = rand_bool()
|
|
car.has_power_steering = rand_bool()
|
|
car.has_cruise_control = rand_bool()
|
|
car.cup_holders = rand_int(12)
|
|
car.has_nav_system = rand_bool()
|
|
|
|
|
|
def calc_value(car):
|
|
result = 0
|
|
|
|
result += car.seats * 200
|
|
result += car.doors * 350
|
|
for wheel in car.wheel:
|
|
result += wheel.diameter * wheel.diameter
|
|
result += 100 if wheel.snow_tires else 0
|
|
|
|
result += car.length * car.width * car.height // 50
|
|
|
|
engine = car.engine
|
|
result += engine.horsepower * 40
|
|
if engine.uses_electric:
|
|
if engine.uses_gas:
|
|
result += 5000
|
|
else:
|
|
result += 3000
|
|
|
|
result += 100 if car.has_power_windows else 0
|
|
result += 200 if car.has_power_steering else 0
|
|
result += 400 if car.has_cruise_control else 0
|
|
result += 2000 if car.has_nav_system else 0
|
|
|
|
result += car.cup_holders * 25
|
|
|
|
return result
|
|
|
|
|
|
class Benchmark:
|
|
def __init__(self, compression):
|
|
self.Request = carsales_pb2.ParkingLot
|
|
self.Response = carsales_pb2.TotalValue
|
|
self.from_bytes_request = from_bytes_helper(carsales_pb2.ParkingLot)
|
|
self.from_bytes_response = from_bytes_helper(carsales_pb2.TotalValue)
|
|
self.to_bytes = lambda x: x.SerializeToString()
|
|
|
|
def setup(self, request):
|
|
result = 0
|
|
for _ in range(rand_int(200)):
|
|
car = request.car.add()
|
|
random_car(car)
|
|
result += calc_value(car)
|
|
return result
|
|
|
|
def handle(self, request, response):
|
|
result = 0
|
|
for car in request.car:
|
|
result += calc_value(car)
|
|
|
|
response.amount = result
|
|
|
|
def check(self, response, expected):
|
|
return response.amount == expected
|