tests.hamcrest_unit_test.collection.BaseQuasiDictionaryIterator

Complexity

Total Complexity

4

Size/Duplication

Total Lines	13
Duplicated Lines	0 %

3 Methods

Rating	Name	Duplication	Size	Complexity
A	BaseQuasiDictionaryIterator.__iter__()	0	2	1
A	BaseQuasiDictionaryIterator.__init__()	0	2	1
A	BaseQuasiDictionaryIterator.__next__()	0	6	2

__author__ = "Jon Reid"
__copyright__ = "Copyright 2011 hamcrest.org"
__license__ = "BSD, see License.txt"

import six


class QuasiDictionary(object):
    def items(self):
        return QuasiDictionaryItemIterator()

    def keys(self):
        return QuasiDictionaryKeyIterator()

    def values(self):
        return QuasiDictionaryValueIterator()


class BaseQuasiDictionaryIterator(six.Iterator):
    def __init__(self):
        self.index = 1

    def __iter__(self):
        return self

    def __next__(self):
        if self.index >= 3:
            raise StopIteration
        result = self.indexToResult()
        self.index += 1
        return result


class QuasiDictionaryItemIterator(BaseQuasiDictionaryIterator):
    def indexToResult(self):
        return (self.index, str(self.index))


class QuasiDictionaryKeyIterator(BaseQuasiDictionaryIterator):
    def indexToResult(self):
        return self.index


class QuasiDictionaryValueIterator(BaseQuasiDictionaryIterator):
    def indexToResult(self):
        return str(self.index)