def test_set_ambient_light_frequency():
from zivid.capture_assistant import SuggestSettingsParameters
suggested_settings = SuggestSettingsParameters()
suggested_settings.ambient_light_frequency = (
SuggestSettingsParameters.AmbientLightFrequency.hz50)
assert (suggested_settings.ambient_light_frequency ==
SuggestSettingsParameters.AmbientLightFrequency.hz50)
assert suggested_settings.ambient_light_frequency == "hz50"
assert isinstance(suggested_settings.ambient_light_frequency, str)
suggested_settings.ambient_light_frequency = (
SuggestSettingsParameters.AmbientLightFrequency.hz60)
assert (suggested_settings.ambient_light_frequency ==
SuggestSettingsParameters.AmbientLightFrequency.hz60)
assert suggested_settings.ambient_light_frequency == "hz60"
assert isinstance(suggested_settings.ambient_light_frequency, str)
suggested_settings.ambient_light_frequency = (
SuggestSettingsParameters.AmbientLightFrequency.none)
assert (suggested_settings.ambient_light_frequency ==
SuggestSettingsParameters.AmbientLightFrequency.none)
assert suggested_settings.ambient_light_frequency == "none"
assert isinstance(suggested_settings.ambient_light_frequency, str)
def test_suggest_settings_parameters():
import datetime
from zivid.capture_assistant import SuggestSettingsParameters
# Use constructor
suggest_settings_parameters = SuggestSettingsParameters(
max_capture_time=datetime.timedelta(milliseconds=1200),
ambient_light_frequency=SuggestSettingsParameters.
AmbientLightFrequency.hz50,
)
assert isinstance(suggest_settings_parameters.max_capture_time,
datetime.timedelta)
assert suggest_settings_parameters.max_capture_time == datetime.timedelta(
seconds=1.2)
assert isinstance(suggest_settings_parameters.ambient_light_frequency, str)
assert (suggest_settings_parameters.ambient_light_frequency ==
SuggestSettingsParameters.AmbientLightFrequency.hz50)
# Use setters
new_time = datetime.timedelta(milliseconds=1800)
new_freq = "hz60"
suggest_settings_parameters.max_capture_time = new_time
assert isinstance(suggest_settings_parameters.max_capture_time,
datetime.timedelta)
assert suggest_settings_parameters.max_capture_time == new_time
suggest_settings_parameters.ambient_light_frequency = new_freq
assert isinstance(suggest_settings_parameters.ambient_light_frequency, str)
assert suggest_settings_parameters.ambient_light_frequency == new_freq
def test_suggest_settings_throws_if_budget_outside_range():
import datetime
from zivid.capture_assistant import SuggestSettingsParameters
# too small
with pytest.raises(IndexError):
SuggestSettingsParameters(
max_capture_time=datetime.timedelta(milliseconds=100),
ambient_light_frequency=SuggestSettingsParameters.
AmbientLightFrequency.hz50,
)
# too big
with pytest.raises(IndexError):
SuggestSettingsParameters(max_capture_time=datetime.timedelta(
milliseconds=60000))
def test_init_max_capture_time(application): # pylint: disable=unused-argument
from zivid.capture_assistant import SuggestSettingsParameters
suggested_settings = SuggestSettingsParameters(datetime.timedelta(milliseconds=100))
max_capture_time = suggested_settings.max_capture_time
assert max_capture_time is not None
assert isinstance(max_capture_time, datetime.timedelta)
assert max_capture_time == datetime.timedelta(milliseconds=100)
def test_init_max_capture_time():
import datetime
from zivid.capture_assistant import SuggestSettingsParameters
suggested_settings = SuggestSettingsParameters(
max_capture_time=datetime.timedelta(milliseconds=1000))
max_capture_time = suggested_settings.max_capture_time
assert max_capture_time is not None
assert isinstance(max_capture_time, datetime.timedelta)
assert max_capture_time == datetime.timedelta(milliseconds=1000)
def test_suggest_settings_throws_if_budget_outside_range(file_camera):
import zivid
from zivid.capture_assistant import AmbientLightFrequency, SuggestSettingsParameters
# too small
suggest_settings_parameters = SuggestSettingsParameters(
max_capture_time=datetime.timedelta(milliseconds=100),
ambient_light_frequency=AmbientLightFrequency.hz50,
)
with pytest.raises(RuntimeError):
zivid.capture_assistant.suggest_settings(file_camera,
suggest_settings_parameters)
# too big
suggest_settings_parameters = SuggestSettingsParameters(
max_capture_time=datetime.timedelta(milliseconds=60000))
with pytest.raises(RuntimeError):
zivid.capture_assistant.suggest_settings(file_camera,
suggest_settings_parameters)
def test_default_ambient_light_frequency(
application # pylint: disable=unused-argument
):
from zivid.capture_assistant import SuggestSettingsParameters, AmbientLightFrequency
suggested_settings = SuggestSettingsParameters(datetime.timedelta(milliseconds=100))
ambient_light_frequency = suggested_settings.ambient_light_frequency
assert ambient_light_frequency is not None
assert isinstance(ambient_light_frequency, AmbientLightFrequency)
assert ambient_light_frequency == AmbientLightFrequency.none
def test_default_ambient_light_frequency():
import datetime
from zivid.capture_assistant import SuggestSettingsParameters
suggested_settings = SuggestSettingsParameters(
max_capture_time=datetime.timedelta(milliseconds=250))
ambient_light_frequency = suggested_settings.ambient_light_frequency
assert ambient_light_frequency is not None
assert isinstance(ambient_light_frequency, str)
assert (ambient_light_frequency ==
SuggestSettingsParameters.AmbientLightFrequency.none)
def _main():
app = zivid.Application()
camera = app.connect_camera()
suggest_settings_parameters = SuggestSettingsParameters(
max_capture_time=datetime.timedelta(milliseconds=1200),
ambient_light_frequency=AmbientLightFrequency.hz50,
)
settings_list = zivid.capture_assistant.suggest_settings(
camera, suggest_settings_parameters)
with zivid.hdr.capture(camera, settings_list) as hdr_frame:
hdr_frame.save("Result.zdf")
def test_suggest_settings_parameters():
from zivid.capture_assistant import AmbientLightFrequency, SuggestSettingsParameters
suggest_settings_parameters = SuggestSettingsParameters(
max_capture_time=datetime.timedelta(milliseconds=1200),
ambient_light_frequency=AmbientLightFrequency.hz50,
)
assert isinstance(suggest_settings_parameters.max_capture_time,
datetime.timedelta)
assert suggest_settings_parameters.max_capture_time == datetime.timedelta(
seconds=1.2)
assert isinstance(suggest_settings_parameters.ambient_light_frequency,
AmbientLightFrequency)
assert (suggest_settings_parameters.ambient_light_frequency ==
AmbientLightFrequency.hz50)
def test_to_internal_suggest_settings_parameters_to_suggest_settings_parameters_default(
):
from zivid.capture_assistant import SuggestSettingsParameters
from zivid._capture_assistant_suggest_settings_parameters_converter import (
to_capture_assistant_suggest_settings_parameters,
to_internal_capture_assistant_suggest_settings_parameters,
)
default_suggest_settings_parameters = SuggestSettingsParameters()
converted_suggest_settings_parameters = to_capture_assistant_suggest_settings_parameters(
to_internal_capture_assistant_suggest_settings_parameters(
default_suggest_settings_parameters))
assert default_suggest_settings_parameters == converted_suggest_settings_parameters
assert isinstance(converted_suggest_settings_parameters,
SuggestSettingsParameters)
assert isinstance(default_suggest_settings_parameters,
SuggestSettingsParameters)
def test_to_internal_suggest_settings_parameters_to_suggest_settings_parameters_modified(
):
from zivid.capture_assistant import SuggestSettingsParameters
from zivid._capture_assistant_suggest_settings_parameters_converter import (
to_capture_assistant_suggest_settings_parameters,
to_internal_capture_assistant_suggest_settings_parameters,
)
modified_suggest_settings_parameters = SuggestSettingsParameters(
ambient_light_frequency=SuggestSettingsParameters.
AmbientLightFrequency.hz50)
converted_suggest_settings_parameters = to_capture_assistant_suggest_settings_parameters(
to_internal_capture_assistant_suggest_settings_parameters(
modified_suggest_settings_parameters))
assert modified_suggest_settings_parameters == converted_suggest_settings_parameters
assert isinstance(converted_suggest_settings_parameters,
SuggestSettingsParameters)
assert isinstance(modified_suggest_settings_parameters,
SuggestSettingsParameters)
def test_ambient_light_frequency_str():
from zivid.capture_assistant import SuggestSettingsParameters
string = str(SuggestSettingsParameters().ambient_light_frequency)
assert string is not None
assert isinstance(string, str)
def test_suggest_settings_str():
from zivid.capture_assistant import SuggestSettingsParameters
string = str(SuggestSettingsParameters())
assert string is not None
assert isinstance(string, str)
