FilesExpand file tree

 master

/

test_capture_assistant.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

140 lines (111 loc) · 5.13 KB

 master

/

test_capture_assistant.py

Copy path

Top

File metadata and controls

• Code
• Blame

140 lines (111 loc) · 5.13 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

import pytest

def test_ambient_light_frequency():

from zivid.capture_assistant import SuggestSettingsParameters

assert str(SuggestSettingsParameters.AmbientLightFrequency.hz50) == "hz50"

assert str(SuggestSettingsParameters.AmbientLightFrequency.hz60) == "hz60"

assert str(SuggestSettingsParameters.AmbientLightFrequency.none) == "none"

assert SuggestSettingsParameters.AmbientLightFrequency.hz50 == "hz50"

assert SuggestSettingsParameters.AmbientLightFrequency.hz60 == "hz60"

assert SuggestSettingsParameters.AmbientLightFrequency.none == "none"

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():

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_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 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_str():

from zivid.capture_assistant import SuggestSettingsParameters

string = str(SuggestSettingsParameters())

assert string is not None

assert isinstance(string, str)

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)