mmgen-wallet/test/unit_tests_d/ut_indexed_dict.py

#!/usr/bin/env python3
"""
test/unit_tests_d/ut_indexed_dict: IndexedDict class unit test for the MMGen suite
"""

from mmgen.common import *

class unit_test(object):

	def run_test(self,name,ut):
		bad_msg = ( 'initializing values via constructor',
					'reassignment to existing key',
					'item deletion',
					'item moving',
					'clearing',
					'updating' )
		def bad0(): IndexedDict(arg)
		def bad1(): d['a'] = 2
		def bad2(): del d['a']
		def bad3(): d.move_to_end('a')
		def bad4(): d.clear()
		def bad5(): d.update(d)

		def odie(n): die(4,f'\nillegal action {bad_msg[n]!r} failed to raise exception')
		def omsg(e): vmsg(' - ' + e.args[0])

		msg_r('Testing class IndexedDict...')

		from mmgen.obj import IndexedDict
		d = IndexedDict()

		d['a'] = 1
		d['b'] = 2

		vmsg('\nChecking error handling:')

		arg = [('a',1),('b',2)]
		dict(arg)

		for i in range(6):
			try: locals()['bad'+str(i)]()
			except NotImplementedError as e: omsg(e)
			else: odie(i)

		try: d.key(2)
		except Exception as e: omsg(e)
		else: odie('list index out of range')

		d['c'] = 3

		d_chk = {'a':1,'b':2,'c':3}
		assert d == d_chk, d

		d_keys_chk = ['a','b','c']
		assert d.keys == d_keys_chk, d.keys

		A = d.key(0)
		assert A == 'a', A

		A = d.key(2)
		assert A == 'c', A

		msg('OK')
		return True