Commit cd692bcb authored by lucas_miranda's avatar lucas_miranda
Browse files

Fixed issue with gmvaep when returning both prior and posterior

parent 88d5bc99
......@@ -8,7 +8,7 @@
<br>
<div align="center">
<img width="200" height="200" src="https://gitlab.mpcdf.mpg.de/lucasmir/deepof/-/raw/master/Logos/deepOF_logo_w_text.png">
<img width="200" height="200" src="https://gitlab.mpcdf.mpg.de/lucasmir/deepof/-/raw/master/logos/deepOF_logo_w_text.png">
</div>
### A suite for postprocessing time-series extracted from videos of freely moving rodents using [DeepLabCut](http://www.mousemotorlab.org/deeplabcut)
......
......@@ -203,6 +203,9 @@ def test_get_table_dicts(nodes, ego, exclude, sampler):
polar = st.one_of(st.just(True), st.just(False))
speed = sampler.draw(st.integers(min_value=0, max_value=5))
propagate = sampler.draw(st.booleans())
propagate_annots = sampler.draw(
st.one_of(st.just(prun.rule_based_annotation()), st.just(False))
)
coords = prun.get_coords(
center=sampler.draw(st.one_of(st.just("arena"), st.just("Center"))),
......@@ -211,6 +214,7 @@ def test_get_table_dicts(nodes, ego, exclude, sampler):
align=algn,
align_inplace=inplace,
propagate_labels=propagate,
propagate_annotations=propagate_annots,
)
speeds = prun.get_coords(
center=sampler.draw(st.one_of(st.just("arena"), st.just("Center"))),
......@@ -218,17 +222,20 @@ def test_get_table_dicts(nodes, ego, exclude, sampler):
length=sampler.draw(st.one_of(st.just(False), st.just("00:10:00"))),
speed=speed,
propagate_labels=propagate,
propagate_annotations=propagate_annots,
)
distances = prun.get_distances(
length=sampler.draw(st.one_of(st.just(False), st.just("00:10:00"))),
speed=sampler.draw(st.integers(min_value=0, max_value=5)),
propagate_labels=propagate,
propagate_annotations=propagate_annots,
)
angles = prun.get_angles(
degrees=sampler.draw(st.booleans()),
length=sampler.draw(st.one_of(st.just(False), st.just("00:10:00"))),
speed=sampler.draw(st.integers(min_value=0, max_value=5)),
propagate_labels=propagate,
propagate_annotations=propagate_annots,
)
# deepof.coordinates testing
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment