Skip to content
Snippets Groups Projects
Commit 5c5b0b67 authored by Piotr Maślanka's avatar Piotr Maślanka
Browse files

fixed unit tests

parent 4326877d
No related branches found
No related tags found
No related merge requests found
...@@ -43,7 +43,7 @@ class DiscreteSeriesBuilder(object): ...@@ -43,7 +43,7 @@ class DiscreteSeriesBuilder(object):
for k, v in self.series.data: for k, v in self.series.data:
if k in cp_new_data: if k in cp_new_data:
v = cp_new_data.pop(k) v = cp_new_data.pop(k)
new_data.append((k, v)) new_data.add((k, v))
# Add those that remained # Add those that remained
for k, v in cp_new_data.items(): for k, v in cp_new_data.items():
......
...@@ -19,7 +19,7 @@ class SeriesBundle(Series): ...@@ -19,7 +19,7 @@ class SeriesBundle(Series):
(p.domain for p in series), (p.domain for p in series),
REAL_SET) REAL_SET)
super(SeriesBundle, self).__init__(domain) super().__init__(domain)
self.series = series self.series = series
...@@ -32,7 +32,7 @@ class DiscreteSeriesBundle(SeriesBundle): ...@@ -32,7 +32,7 @@ class DiscreteSeriesBundle(SeriesBundle):
""" """
:raise TypeError: if not all series are discrete :raise TypeError: if not all series are discrete
""" """
super(DiscreteSeriesBundle, self).__init__(*series) super().__init__(*series)
if any((not isinstance(s, DiscreteSeries)) for s in series): if any((not isinstance(s, DiscreteSeries)) for s in series):
raise TypeError('All series must be discrete') raise TypeError('All series must be discrete')
...@@ -57,6 +57,6 @@ class DiscreteSeriesBundle(SeriesBundle): ...@@ -57,6 +57,6 @@ class DiscreteSeriesBundle(SeriesBundle):
if len(data) > 0 and data[-1][1] == v: if len(data) > 0 and data[-1][1] == v:
continue continue
else: else:
data.append((k, v)) data.add((k, v))
return DiscreteSeries([(k, self._get_for(k)) for k in keys], self.domain) return DiscreteSeries([(k, self._get_for(k)) for k in keys], self.domain)
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