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))