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(), [])