Python Result._get_cacheの例

コード例 #1

ファイル: test_result.py プロジェクト: shawnbrown/squint

    def test_tuple(self):
        result = Result(iter([1, 2, 3, 4]), evaltype=tuple)

        self.assertEqual(result._get_cache(), ())

        result._next_cache()
        self.assertEqual(result._get_cache(), (1, ))

        result._next_cache()
        result._next_cache()
        result._next_cache()
        self.assertEqual(result._get_cache(), (1, 2, 3, 4))

        with self.assertRaises(StopIteration):
            result._next_cache()

        self.assertEqual(result.fetch(), (1, 2, 3, 4))

コード例 #2

ファイル: test_result.py プロジェクト: shawnbrown/squint

    def test_mapping(self):
        iterable = IterItems([
            ('a', Result(iter([1, 2]), list)),
            ('b', Result(iter([3, 4]), list)),
            ('c', Result(iter([5, 6]), list)),
        ])
        result = Result(iterable, dict)

        self.assertEqual(result._get_cache(), {})

        result._next_cache()
        self.assertEqual(result._cache[0][0], 'a')
        self.assertEqual(result._cache[0][1]._cache[0], 1)
        self.assertEqual(result._get_cache(), {'a': [1]})

        result._next_cache()
        self.assertEqual(result._get_cache(), {'a': [1, 2]})

        result._next_cache()
        self.assertEqual(result._get_cache(), {'a': [1, 2], 'b': [3]})

        result._next_cache()
        result._next_cache()
        result._next_cache()
        self.assertEqual(result._get_cache(), {
            'a': [1, 2],
            'b': [3, 4],
            'c': [5, 6]
        })

        with self.assertRaises(StopIteration):
            result._next_cache()

        self.assertEqual(result.fetch(), {
            'a': [1, 2],
            'b': [3, 4],
            'c': [5, 6]
        })