Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Bliss
python-handel
Commits
862b6e16
Commit
862b6e16
authored
Sep 13, 2017
by
Vincent Michel
Browse files
Update tests
parent
0c202935
Pipeline
#968
failed with stages
in 1 minute and 6 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
tests/test_interface.py
View file @
862b6e16
...
...
@@ -134,6 +134,14 @@ def test_start_run(interface):
m
.
assert_called_once_with
(
2
,
True
)
# Make sure errors have been checked
interface
.
check_error
.
assert_called_once_with
(
0
)
# Reset
m
.
reset_mock
()
interface
.
check_error
.
reset_mock
()
# Third test
assert
interface
.
start_run
()
is
None
m
.
assert_called_once_with
(
-
1
,
False
)
# Make sure errors have been checked
interface
.
check_error
.
assert_called_once_with
(
0
)
def
test_stop_run
(
interface
):
...
...
@@ -153,6 +161,14 @@ def test_stop_run(interface):
m
.
assert_called_once_with
(
2
)
# Make sure errors have been checked
interface
.
check_error
.
assert_called_once_with
(
0
)
# Reset
m
.
reset_mock
()
interface
.
check_error
.
reset_mock
()
# Second test
assert
interface
.
stop_run
()
is
None
m
.
assert_called_once_with
(
-
1
)
# Make sure errors have been checked
interface
.
check_error
.
assert_called_once_with
(
0
)
def
test_get_spectrum_length
(
interface
):
...
...
@@ -195,6 +211,85 @@ def test_get_spectrum(interface):
interface
.
check_error
.
assert_called_with
(
0
)
def
test_get_spectrums
(
interface
):
with
mock
.
patch
(
'handel.interface.get_channels'
)
as
m1
:
with
mock
.
patch
(
'handel.interface.get_spectrum'
)
as
m2
:
m1
.
return_value
=
[
0
,
1
]
# First test
spectrums
=
[[
0
,
1
,
2
],
[
2
,
3
,
4
]]
m2
.
side_effect
=
lambda
x
:
spectrums
[
x
]
expected
=
{
0
:
[
0
,
1
,
2
],
1
:
[
2
,
3
,
4
]}
assert
interface
.
get_spectrums
()
==
expected
m1
.
assert_called_once_with
()
m2
.
assert_called_with
(
1
)
def
test_is_channel_running
(
interface
):
m
=
interface
.
handel
.
xiaGetRunData
def
side_effect
(
channel
,
dtype
,
arg
):
arg
[
0
]
=
1
return
0
m
.
side_effect
=
side_effect
assert
interface
.
is_channel_running
(
2
)
is
True
arg
=
m
.
call_args
[
0
][
2
]
m
.
assert_called_once_with
(
2
,
b
'run_active'
,
arg
)
# Make sure errors have been checked
interface
.
check_error
.
assert_called_with
(
0
)
def
test_is_running
(
interface
):
with
mock
.
patch
(
'handel.interface.get_channels'
)
as
m1
:
with
mock
.
patch
(
'handel.interface.is_channel_running'
)
as
m2
:
m1
.
return_value
=
[
0
,
1
,
2
]
# First test
m2
.
side_effect
=
lambda
x
:
x
==
1
assert
interface
.
is_running
()
is
True
m1
.
assert_called_once_with
()
m2
.
assert_called_with
(
1
)
# Second test
m2
.
side_effect
=
lambda
x
:
False
assert
interface
.
is_running
()
is
False
# Statistics
def
test_get_module_statistics
(
interface
):
m
=
interface
.
handel
.
xiaGetRunData
def
side_effect
(
channel
,
dtype
,
arg
):
arg
[
3
*
7
:
4
*
7
]
=
[
1.0
,
2.0
,
255
,
3
,
4
,
8
,
6
]
return
0
m
.
side_effect
=
side_effect
with
mock
.
patch
(
'handel.interface.get_module_channels'
)
as
m2
:
m2
.
return_value
=
[
-
1
,
-
1
,
-
1
,
8
]
expected
=
{
8
:
interface
.
Stats
(
1.0
,
2.0
,
3
,
4
,
8
,
6
,
0.25
)}
assert
interface
.
get_module_statistics
(
'module3'
)
==
expected
m2
.
assert_called_once_with
(
'module3'
)
arg
=
m
.
call_args
[
0
][
2
]
m
.
assert_called_once_with
(
8
,
b
'module_statistics'
,
arg
)
# Make sure errors have been checked
interface
.
check_error
.
assert_called_with
(
0
)
def
test_get_statistics
(
interface
):
with
mock
.
patch
(
'handel.interface.get_modules'
)
as
m1
:
with
mock
.
patch
(
'handel.interface.get_module_statistics'
)
as
m2
:
m1
.
return_value
=
[
'module1'
,
'module2'
]
# First test
stats
=
{
'module1'
:
{
0
:
'some'
,
1
:
'stats'
},
'module2'
:
{
2
:
'and'
,
3
:
'some more'
}}
m2
.
side_effect
=
lambda
x
:
stats
[
x
]
expected
=
{
0
:
'some'
,
1
:
'stats'
,
2
:
'and'
,
3
:
'some more'
}
assert
interface
.
get_statistics
()
==
expected
m1
.
assert_called_once_with
()
m2
.
assert_called_with
(
'module2'
)
# Buffer
def
test_get_buffer_length
(
interface
):
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment