def test_get_targets_as_dict_with_indexes_list(self): req_idx = [0, 3] exp_targets = self._create_targets(req_idx) holder = targ.TargetsHolder() holder.extend(self._create_targets(slice(None))) self._assertTargetsDictEqual(holder.get_targets_as_dict(req_idx), exp_targets)
def test_get_targets_as_dict_with_range(self): req_range = range(1, 3) exp_targets = self._create_targets(req_range) holder = targ.TargetsHolder() holder.extend(self._create_targets(slice(None))) self._assertTargetsDictEqual(holder.get_targets_as_dict(req_range), exp_targets)
def test_get_targets_as_arraylike_with_indexes_list(self): req_idx = [0, 2] targets_dict = self._create_targets(req_idx) exp_targets = (targets_dict['classification'], ) holder = targ.TargetsHolder() holder.extend(self._create_targets(slice(None))) self._assertTargetsArraylikeEqual( holder.get_targets_as_arraylike(req_idx), exp_targets)
def test_get_targets_as_arraylike_with_range(self): req_range = range(1, 3) targets_dict = self._create_targets(req_range) exp_targets = (targets_dict['classification'], ) holder = targ.TargetsHolder() holder.extend(self._create_targets(slice(None))) self._assertTargetsArraylikeEqual( holder.get_targets_as_arraylike(req_range), exp_targets)
def test_get_targets_as_arraylike_with_slice(self): req_slice = slice(0, 2) targets = self._create_targets(req_slice) exp_targets = (targets['classification'], ) holder = targ.TargetsHolder() holder.extend(self._create_targets(slice(None))) self._assertTargetsArraylikeEqual( holder.get_targets_as_arraylike(req_slice), exp_targets)
def __init__(self, name, packet_shape, resizable=True, dtype=np.uint8, item_types={'raw': True, 'yx': False, 'gtux': False, 'gtuy': False}): self._data = dat.DataHolder(packet_shape, item_types=item_types, dtype=dtype) self._targ = targ.TargetsHolder() self._meta = meta.MetadataHolder() self._num_data = 0 self.resizable = True self.name = name
def test_shuffle(self): holder = targ.TargetsHolder() targets = {'classification': [[0, 0], [0, 1], [1, 0], [1, 1]]} holder.extend(targets) exp_targets = targets.copy() def shuffler(seq): temp = seq[0] seq[0] = seq[1] seq[1] = temp shuffler(exp_targets['classification']) holder.shuffle(shuffler, lambda: None) self._assertTargetsDictEqual(holder.get_targets_as_dict(), exp_targets)
def test_get_targets_as_arraylike_empty(self): holder = targ.TargetsHolder() self._assertTargetsArraylikeEqual(holder.get_targets_as_arraylike(), ([], ))
def test_get_targets_as_dict_empty(self): holder = targ.TargetsHolder() self._assertTargetsDictEqual(holder.get_targets_as_dict(), {'classification': [], })
def test_length_with_items(self): holder = targ.TargetsHolder() targets = self._create_targets(slice(1, 4)) holder.extend(targets) self.assertEqual(len(holder), 3)
def test_length_empty(self): holder = targ.TargetsHolder() self.assertEqual(len(holder), 0)
def test_append(self): target = self._create_targets(0) exp_target = {'classification': [target['classification']]} holder = targ.TargetsHolder() holder.append(target) self._assertTargetsDictEqual(holder.get_targets_as_dict(), exp_target)
def test_extend(self): targets = self._create_targets(slice(None)) exp_targets = {'classification': targets['classification']} holder = targ.TargetsHolder() holder.extend(targets) self._assertTargetsDictEqual(holder.get_targets_as_dict(), exp_targets)