def test_saving_calibrations(extension, make_temp_file):
original_params = get_fake_calibration_parameters(200)
calibrations_file = Path(make_temp_file("." + extension))
save_calibrations(original_params, calibrations_file)
read_params = parse_calibration_file(calibrations_file)
assert read_params[0].tolist() == original_params[0].tolist()
assert read_params[1].tolist() == original_params[1].tolist()
def __init__(
self,
marker_id: int,
marker_size: int,
*,
marker_type: MarkerType,
border_size: int = 40,
) -> None:
if marker_id > marker_type.max_id:
raise ValueError(
"marker id {} must be less than the maximum allowed by {}: {}".
format(marker_id, marker_type.name, marker_type.max_id))
if marker_size < marker_type.min_marker_image_size:
raise ValueError("marker must be at least {} pixels wide".format(
marker_type.min_marker_image_size))
if border_size < self.MIN_BORDER_SIZE:
raise ValueError("Border size must be at least {}".format(
self.MIN_BORDER_SIZE))
super().__init__(
marker_size=marker_size,
marker_type=marker_type,
)
self.marker_id = marker_id
self.border_size = border_size
self.calibration_params = get_fake_calibration_parameters()
def fake_calibration_params():
return get_fake_calibration_parameters(200)
def get_calibrations(self):
return get_fake_calibration_parameters(self.marker_size)
def fake_calibration_params() -> CalibrationParameters:
return get_fake_calibration_parameters(200)
def test_cant_save_invalid_extension():
with pytest.raises(ValueError) as e:
save_calibrations(get_fake_calibration_parameters(200),
Path("test.unknown"))
assert "Unknown calibration file format" in e.value.args[0]
def get_calibrations(self) -> Optional[CalibrationParameters]:
return get_fake_calibration_parameters(self.marker_size)