You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
989 B
Python
38 lines
989 B
Python
from tests.base import main
|
|
from src.eynollah.utils.counter import EynollahIdCounter
|
|
|
|
|
|
def test_counter_string():
|
|
c = EynollahIdCounter()
|
|
assert c.next_region_id == 'region_0001'
|
|
assert c.next_region_id == 'region_0002'
|
|
assert c.next_line_id == 'region_0002_line_0001'
|
|
assert c.next_region_id == 'region_0003'
|
|
assert c.next_line_id == 'region_0003_line_0001'
|
|
assert c.region_id(999) == 'region_0999'
|
|
assert c.line_id(999, 888) == 'region_0999_line_0888'
|
|
|
|
|
|
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()
|
|
assert c.get('region') == 0
|
|
c.inc('region', 5)
|
|
assert c.get('region') == 5
|
|
c.set('region', 10)
|
|
assert c.get('region') == 10
|
|
c.inc('region', -9)
|
|
assert c.get('region') == 1
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main(__file__)
|