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
beat
beat.cmdline
Commits
774858ae
Commit
774858ae
authored
Oct 22, 2019
by
Samuel GAIST
Browse files
[test][experiments] Add fork push test setup
parent
6f82e8cc
Changes
1
Hide whitespace changes
Inline
Side-by-side
beat/cmdline/test/test_experiments.py
View file @
774858ae
...
...
@@ -45,13 +45,17 @@ from beat.core.test.test_execution import create_conda_environment
from
beat.core.test.test_execution
import
clear_conda_environment
from
beat.core.experiment
import
Storage
,
Experiment
from
beat.core.toolchain
import
Storage
as
TCStorage
from
beat.core.algorithm
import
Storage
as
AlgStorage
from
beat.core.dataformat
import
Storage
as
DFStorage
from
beat.core.database
import
Storage
as
DBStorage
from
..common
import
Selector
from
.utils
import
index_experiment_dbs
,
MockLoggingHandler
from
.
import
core
from
.
import
tmp_prefix
from
.
import
prefix
,
tmp_prefix
def
setup_experiments
():
...
...
@@ -67,7 +71,8 @@ class TestOnlineExperiments(core.OnlineAssetTestCase):
"pull"
:
"user/user/single/1/single"
,
"diff"
:
"user/user/single/1/single"
,
"push"
:
"user/user/unknown/1/unknown"
,
"push"
:
"user/user/unknown/1/unknown"
,
"fork_from"
:
"user/user/unknown/1/unknown"
,
"fork"
:
"user/user/unknown/1/forked_obj"
,
"not_owner_push"
:
"other_user/other_user/somechain/1/someexp"
,
}
...
...
@@ -78,6 +83,23 @@ class TestOnlineExperiments(core.OnlineAssetTestCase):
experiment
.
data
[
"globals"
][
"queue"
]
=
"another_queue"
experiment
.
write
()
def
_prepare_fork_dependencies
(
self
,
asset_name
):
super
().
_prepare_fork_dependencies
(
asset_name
)
assets
=
[
(
DBStorage
,
[
"simple/1"
]),
(
DFStorage
,
[
"user/single_integer/1"
]),
(
AlgStorage
,
[
"user/unknown/1"
,
"user/integers_echo_analyzer_v2/1"
]),
(
TCStorage
,
[
"user/unknown/1"
]),
(
Storage
,
[
"user/user/unknown/1/unknown"
]),
]
for
item
in
assets
:
storage_cls
,
asset_list
=
item
for
asset
in
asset_list
:
src_storage
=
storage_cls
(
prefix
,
asset
)
dst_storage
=
storage_cls
(
tmp_prefix
,
asset
)
dst_storage
.
save
(
*
src_storage
.
load
())
@
core
.
skip_disconnected
def
test_fork
(
self
,
obj
=
None
,
obj2
=
None
):
obj
=
obj
or
"user/user/single/1/single"
...
...
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