d0df704d8a
added python.path vm arg to startup script fixed infinite loop in unwrap() when displaying sequences of sequences git-svn-id: http://google-refine.googlecode.com/svn/trunk@509 7d457c2a-affb-35e4-300a-418c747d4874
37 lines
932 B
Python
37 lines
932 B
Python
"""Misc iterator tests.
|
|
|
|
Made for Jython.
|
|
"""
|
|
import itertools
|
|
from test import test_support
|
|
import unittest
|
|
|
|
class IterTestCase(unittest.TestCase):
|
|
|
|
def test_fastiter(self):
|
|
class MyList(list):
|
|
def __getitem__(self, index):
|
|
return str(index) + '!'
|
|
class MyTuple(tuple):
|
|
def __getitem__(self, index):
|
|
return str(index) + '!'
|
|
self.assertEqual(iter(MyList(['a', 'b'])).next(), 'a')
|
|
self.assertEqual(iter(MyTuple(['a', 'b'])).next(), 'a')
|
|
|
|
def test_slowiter(self):
|
|
class MyStr(str):
|
|
def __getitem__(self, index):
|
|
return str(index) + '!'
|
|
self.assertEqual(iter(MyStr('ab')).next(), '0!')
|
|
|
|
def test_chain(self):
|
|
self.assertEqual(list(itertools.chain([], [], ['foo'])), ['foo'])
|
|
|
|
|
|
def test_main():
|
|
test_support.run_unittest(IterTestCase)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test_main()
|