Skip to content
Snippets Groups Projects
Commit b9f7a90b authored by Alán Muñoz's avatar Alán Muñoz
Browse files

add tests

Former-commit-id: 1115de5f4cfc796e8014e22be8b0758449e63545
parent 6510ead6
No related branches found
No related tags found
No related merge requests found
...@@ -168,7 +168,7 @@ def rec_bottom(d, k): ...@@ -168,7 +168,7 @@ def rec_bottom(d, k):
return rec_bottom(d, d[k]) return rec_bottom(d, d[k])
def join_tracks(tracks, joinable_pairs, drop=False) -> pd.DataFrame: def join_tracks(tracks, joinable_pairs, drop=True) -> pd.DataFrame:
""" """
Join pairs of tracks from later tps towards the start. Join pairs of tracks from later tps towards the start.
...@@ -186,7 +186,7 @@ def join_tracks(tracks, joinable_pairs, drop=False) -> pd.DataFrame: ...@@ -186,7 +186,7 @@ def join_tracks(tracks, joinable_pairs, drop=False) -> pd.DataFrame:
tmp = copy(tracks) tmp = copy(tracks)
for target, source in joinable_pairs: for target, source in joinable_pairs:
tmp.loc[target] = join_track_pairs(tmp.loc[target], tmp.loc[source]) tmp.loc[target] = join_track_pair(tmp.loc[target], tmp.loc[source])
if drop: if drop:
tmp = tmp.drop(source) tmp = tmp.drop(source)
...@@ -194,9 +194,6 @@ def join_tracks(tracks, joinable_pairs, drop=False) -> pd.DataFrame: ...@@ -194,9 +194,6 @@ def join_tracks(tracks, joinable_pairs, drop=False) -> pd.DataFrame:
return tmp return tmp
from copy import copy
def join_track_pair(target, source): def join_track_pair(target, source):
tgt_copy = copy(target) tgt_copy = copy(target)
end = find_1st(target.values[::-1], 0, cmp_larger) end = find_1st(target.values[::-1], 0, cmp_larger)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment