Skip to content
Snippets Groups Projects
Commit 5fe84793 authored by leufen1's avatar leufen1
Browse files

fixed iterator bug when indexing causing the nan error during postprocessing

parent 636ce543
No related branches found
No related tags found
3 merge requests!226Develop,!225Resolve "release v1.2.0",!204Resolve "BUG: NaN error during PostProcessing"
Pipeline #54525 passed
......@@ -55,7 +55,7 @@ class DataCollection(Iterable):
def add(self, element):
self._collection.append(element)
self._mapping[str(element)] = len(self._collection)
self._mapping[str(element)] = len(self._collection) - 1
def _set_mapping(self):
for i, e in enumerate(self._collection):
......
......@@ -52,6 +52,13 @@ class TestDataCollection:
for e, i in enumerate(data_collection):
assert i == e
def test_add(self):
data_collection = DataCollection()
data_collection.add("first_element")
assert len(data_collection) == 1
assert data_collection["first_element"] == "first_element"
assert data_collection[0] == "first_element"
class DummyData:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment