Skip to content
Snippets Groups Projects
Commit 79b7432b authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[test] attempt to figure out whats going on with map_labels

parent 42df0fd6
Branches
Tags v0.0.5
1 merge request!9Light cnn
Pipeline #26409 passed
......@@ -109,17 +109,17 @@ def test_map_labels():
labels = ['1', '4', '7']
from ..datasets import map_labels
new_labels = map_labels(labels)
assert '0' in new_labels
assert '1' in new_labels
assert '2' in new_labels
assert '0' in new_labels, "new_labels = {}".format(new_labels)
assert '1' in new_labels, "new_labels = {}".format(new_labels)
assert '2' in new_labels, "new_labels = {}".format(new_labels)
#new_labels = sorted(new_labels)
#assert new_labels == ['0', '1', '2']
new_labels = map_labels(labels, start_index = 5)
#new_labels = sorted(new_labels)
assert '5' in new_labels
assert '6' in new_labels
assert '7' in new_labels
assert '5' in new_labels, "new_labels = {}".format(new_labels)
assert '6' in new_labels, "new_labels = {}".format(new_labels)
assert '7' in new_labels, "new_labels = {}".format(new_labels)
#assert new_labels == ['5', '6', '7']
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment