Light cnn

bob/learn/pytorch/test/test.py

@@ -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']