def test_repr(self):
assert repr(self.query) == u"Query(<lambda>)"
def integers():
"""
Return the first 100 integers
"""
return list(range(100))
assert repr(Query(integers)) == u"Query(integers)"
assert repr(self.query.map(
lambda x: x + 1)) == u"Query(<lambda>).map(<lambda>)"
assert repr(self.query.map(lambda x: x + 1,
'x + 1')) == u"Query(<lambda>).map(x + 1)"
self.assertEqual(
u"Query(<lambda>).map(x + 1).filter(<lambda>)",
repr(
self.query.map(lambda x: x + 1,
'x + 1').filter(lambda x: x > 2)))
self.assertEqual(
u"Query(<lambda>).transform(<lambda>)",
repr(self.query.transform(lambda xs: iter(xs).next(0))))
self.assertEqual(u"Query(<lambda>).filter(text='foo')",
repr(self.query.filter(text='foo')))
def test_repr(self):
self.assertEquals(u"Query(<lambda>)", repr(self.query))
def integers():
return range(100)
self.assertEquals(u"Query(integers)", repr(Query(integers)))
self.assertEquals(u"Query(<lambda>).map(<lambda>)", repr(self.query.map(lambda x: x + 1)))
self.assertEquals(u"Query(<lambda>).map(x + 1)", repr(self.query.map(lambda x: x + 1, 'x + 1')))
self.assertEquals(
u"Query(<lambda>).map(x + 1).filter(<lambda>)",
repr(self.query.map(lambda x: x + 1, 'x + 1').filter(lambda x: x > 2))
)
self.assertEquals(
u"Query(<lambda>).transform(<lambda>)",
repr(self.query.transform(lambda xs: iter(xs).next(0)))
)
self.assertEquals(
u"Query(<lambda>).filter(text='foo')",
repr(self.query.filter(text='foo'))
)
def test_retry_on_error(self):
seed = Mock()
seed.side_effect = [WebDriverException, ["success"]]
self.assertEqual(["success"], Query(seed_fn=seed).results)
def setUp(self):
super(TestQuery, self).setUp()
self.query = Query(lambda: list(range(5)))
def test_nth(self):
query = Query(lambda: list(range(2)))
self.assertEqual([], query.nth(-1).results)
self.assertEqual([0], query.nth(0).results)
self.assertEqual([1], query.nth(1).results)
self.assertEqual([], query.nth(2).results)
def test_first_no_results(self):
query = Query(lambda: [])
self.assertEqual([], query.first.results)
def test_first(self):
query = Query(lambda: list(range(2)))
self.assertEqual([0], query.first.results)
self.assertEqual([0], query.first.first.results)
def setUp(self):
self.query = Query(lambda: range(5))
