def test_unique_scalars_many(self):
result = self._fixture()
result = result.unique().scalars(1)
eq_(result.fetchmany(2), [1, 3])
eq_(result.fetchmany(2), [])
def test_unique_scalars_many_none(self):
result = self._fixture()
result = result.unique().scalars(1)
# this assumes the default fetchmany behavior of all() for
# the ListFetchStrategy
eq_(result.fetchmany(None), [1, 3])
eq_(result.fetchmany(None), [])
def test_unique_filtered_many(self):
result = self._fixture()
def uniq(row):
return row[0]
result = result.unique(uniq)
eq_(result.fetchmany(2), [(1, 1, 1), (2, 1, 2)])
eq_(result.fetchmany(2), [(4, 1, 2)])
eq_(result.fetchmany(2), [])
def test_scalar_none_many(self):
result = self._fixture(data=[
(1, None, 2),
(3, 4, 5),
(3, None, 5),
(3, None, 5),
(3, 4, 5),
])
result = result.scalars(1)
eq_(result.fetchmany(3), [None, 4, None])
eq_(result.fetchmany(5), [None, 4])
def test_many_then_all(self):
result = self._fixture()
eq_(result.fetchmany(3), [(1, 1, 1), (2, 1, 2), (1, 3, 2)])
eq_(result.all(), [(4, 1, 2)])
eq_(result.all(), [])
