Mercurial > piecrust2

comparison tests/test_dataproviders_pageiterator.py @ 1101:31113d52e8be

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
tests: Make internal test stuff not get picked up by pytest.
author Ludovic Chabant <ludovic@chabant.com>
date Sat, 17 Feb 2018 15:08:34 -0800
parents 45ad976712ec
children
comparison
equal deleted inserted replaced
1100:1ce67d2fae0a 1101:31113d52e8be
41 assert it.total_count == 5 41 assert it.total_count == 5
42 assert len(it) == 5 42 assert len(it) == 5
43 assert list(it) == list(reversed(range(5))) 43 assert list(it) == list(reversed(range(5)))
44 44
45 45
46 class TestItem(object): 46 class _TestItem(object):
47 def __init__(self, value): 47 def __init__(self, value):
48 self.name = str(value) 48 self.name = str(value)
49 self.config = {'foo': value} 49 self.config = {'foo': value}
50 50
51 def __eq__(self, other): 51 def __eq__(self, other):
52 return other.name == self.name 52 return other.name == self.name
53 53
54 54
55 def test_setting_sort(): 55 def test_setting_sort():
56 it = PageIterator([TestItem(v) for v in [4, 3, 1, 2, 0]]) 56 it = PageIterator([_TestItem(v) for v in [4, 3, 1, 2, 0]])
57 it.sort('foo') 57 it.sort('foo')
58 assert it.total_count == 5 58 assert it.total_count == 5
59 assert len(it) == 5 59 assert len(it) == 5
60 assert list(it) == [TestItem(v) for v in range(5)] 60 assert list(it) == [_TestItem(v) for v in range(5)]
61 61
62 62
63 def test_setting_sort_reversed(): 63 def test_setting_sort_reversed():
64 it = PageIterator([TestItem(v) for v in [4, 3, 1, 2, 0]]) 64 it = PageIterator([_TestItem(v) for v in [4, 3, 1, 2, 0]])
65 it.sort('foo', reverse=True) 65 it.sort('foo', reverse=True)
66 assert it.total_count == 5 66 assert it.total_count == 5
67 assert len(it) == 5 67 assert len(it) == 5
68 assert list(it) == [TestItem(v) for v in reversed(range(5))] 68 assert list(it) == [_TestItem(v) for v in reversed(range(5))]
69 69
70 70
71 def test_filter(): 71 def test_filter():
72 page = mock.MagicMock(spec=Page) 72 page = mock.MagicMock(spec=Page)
73 page.config = PageConfiguration() 73 page.config = PageConfiguration()
74 page.config.set('threes', {'is_foo': 3}) 74 page.config.set('threes', {'is_foo': 3})
75 it = PageIterator([TestItem(v) for v in [3, 2, 3, 1, 4, 3]], 75 it = PageIterator([_TestItem(v) for v in [3, 2, 3, 1, 4, 3]],
76 current_page=page) 76 current_page=page)
77 it.filter('threes') 77 it.filter('threes')
78 assert it.total_count == 3 78 assert it.total_count == 3
79 assert len(it) == 3 79 assert len(it) == 3
80 assert list(it) == [TestItem(3), TestItem(3), TestItem(3)] 80 assert list(it) == [_TestItem(3), _TestItem(3), _TestItem(3)]
81 81
82 82
83 def test_magic_filter(): 83 def test_magic_filter():
84 it = PageIterator([TestItem(v) for v in [3, 2, 3, 1, 4, 3]]) 84 it = PageIterator([_TestItem(v) for v in [3, 2, 3, 1, 4, 3]])
85 it.is_foo(3) 85 it.is_foo(3)
86 assert it.total_count == 3 86 assert it.total_count == 3
87 assert len(it) == 3 87 assert len(it) == 3
88 assert list(it) == [TestItem(3), TestItem(3), TestItem(3)] 88 assert list(it) == [_TestItem(3), _TestItem(3), _TestItem(3)]
89 89