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(), [])
def test_first(self):
result = self._fixture()
row = result.first()
eq_(row, (1, 1, 1))
eq_(result.all(), [])
def test_alt_row_fetch(self):
class AppleRow(Row):
def apple(self):
return "apple"
result = self._fixture(alt_row=AppleRow)
row = result.all()[0]
eq_(row.apple(), "apple")
def test_partition_yield_per(self):
result = self._fixture()
r = []
for partition in result.yield_per(2).partitions():
r.append(list(partition))
eq_(r, [[(1, 1, 1), (2, 1, 2)], [(1, 3, 2), (4, 1, 2)]])
eq_(result.all(), [])
def test_merge_preserve_unique(self, dupe_fixture):
r1, r2 = dupe_fixture
r1.unique().scalars("y")
r2.unique().scalars("y")
result = r1.merge(r2)
# unique takes place
eq_(result.all(), [2, 1, 3])
def test_merge_scalars(self, merge_fixture):
r1, r2, r3, r4 = merge_fixture
for r in (r1, r2, r3, r4):
r.scalars(0)
result = r1.merge(r2, r3, r4)
eq_(result.all(), [7, 8, 9, 10, 11, 12])
def test_scalar_none_all_unique(self):
result = self._fixture(data=[
(1, None, 2),
(3, 4, 5),
(3, None, 5),
(3, None, 5),
(3, 4, 5),
])
result = result.scalars(1).unique()
eq_(result.all(), [None, 4])
def test_columns(self):
result = self._fixture()
result = result.columns("b", "c")
eq_(result.keys(), ["b", "c"])
eq_(result.all(), [(1, 1), (1, 2), (3, 2), (1, 2)])
def test_scalar(self):
result = self._fixture()
eq_(result.scalar(), 1)
eq_(result.all(), [])
def test_all(self):
result = self._fixture()
eq_(result.all(), [(1, 1, 1), (2, 1, 2), (1, 3, 2), (4, 1, 2)])
eq_(result.all(), [])
