qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-07-22 01:09:56 +02:00
Code Issues Projects Releases Wiki Activity

counter: add reset();

Browse source
This commit is contained in:
Konstantin Baierer 2021-03-12 18:39:27 +01:00
parent a3465ca1a0
commit a678bbf966
2 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
qurator/eynollah/utils/counter.py
View file

@ -7,8 +7,13 @@ class EynollahIdCounter():
def __init__(self, region_idx=0, line_idx=0):
self._counter = Counter()
self.set('region', region_idx)
self.set('line', line_idx)
self._inital_region_idx = region_idx
self._inital_line_idx = line_idx
self.reset()
def reset(self):
self.set('region', self._inital_region_idx)
self.set('line', self._inital_line_idx)
def inc(self, name, val=1):
self._counter.update({name: val})

4
tests/test_counter.py
View file

@ -14,6 +14,10 @@ def test_counter_string():
def test_counter_init():
c = EynollahIdCounter(region_idx=2)
assert c.get('region') == 2
c.inc('region')
assert c.get('region') == 3
c.reset()
assert c.get('region') == 2
def test_counter_methods():
c = EynollahIdCounter()