Skip to content
Snippets Groups Projects
Commit 17f77d5a authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Break the link between SampleSet and Sample

parent 03f9de2a
No related branches found
No related tags found
1 merge request!20Get SampleSet to accept DelayedSample
Pipeline #38738 failed
......@@ -93,18 +93,11 @@ class SampleSet(MutableSequence):
return self.samples.__getitem__(item)
def __setitem__(self, key, item):
if not isinstance(item, Sample) and not isinstance(item, DelayedSample):
raise ValueError(f"item should be of type Sample, not {item}")
return self.samples.__setitem__(key, item)
def __delitem__(self, item):
return self.samples.__delitem__(item)
def insert(self, index, item):
if not isinstance(item, Sample) and not isinstance(item, DelayedSample):
raise ValueError(f"item should be of type Sample, not {item}")
# if not item in self.samples:
self.samples.insert(index, item)
from bob.pipelines.sample import Sample, SampleSet, DelayedSample
import numpy
from nose.tools import assert_raises
import copy
......@@ -23,17 +22,9 @@ def test_sampleset_collection():
del sampleset[0]
assert len(sampleset) == n_samples
# Testing exception
with assert_raises(ValueError):
sampleset.insert(1, 10)
# Testing set
sampleset[0] = copy.deepcopy(sample)
# Testing exception
with assert_raises(ValueError):
sampleset[0] = "xuxa"
# Testing iterator
for i in sampleset:
assert isinstance(i, Sample)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment