From 5c5b0b673387734110b365734ddfef1b94286250 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Piotr=20Ma=C5=9Blanka?= <piotr.maslanka@henrietta.com.pl>
Date: Fri, 7 Feb 2020 14:38:17 +0100
Subject: [PATCH] fixed unit tests
---
firanka/builders.py | 2 +-
firanka/series/bundle.py | 6 +++---
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/firanka/builders.py b/firanka/builders.py
index dea2b20..84ac31b 100644
--- a/firanka/builders.py
+++ b/firanka/builders.py
@@ -43,7 +43,7 @@ class DiscreteSeriesBuilder(object):
for k, v in self.series.data:
if k in cp_new_data:
v = cp_new_data.pop(k)
- new_data.append((k, v))
+ new_data.add((k, v))
# Add those that remained
for k, v in cp_new_data.items():
diff --git a/firanka/series/bundle.py b/firanka/series/bundle.py
index d73920c..999d358 100644
--- a/firanka/series/bundle.py
+++ b/firanka/series/bundle.py
@@ -19,7 +19,7 @@ class SeriesBundle(Series):
(p.domain for p in series),
REAL_SET)
- super(SeriesBundle, self).__init__(domain)
+ super().__init__(domain)
self.series = series
@@ -32,7 +32,7 @@ class DiscreteSeriesBundle(SeriesBundle):
"""
: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):
raise TypeError('All series must be discrete')
@@ -57,6 +57,6 @@ class DiscreteSeriesBundle(SeriesBundle):
if len(data) > 0 and data[-1][1] == v:
continue
else:
- data.append((k, v))
+ data.add((k, v))
return DiscreteSeries([(k, self._get_for(k)) for k in keys], self.domain)
--
GitLab