def testSampleKeyPairEverything(self): ds = bigtable_api._BigtableSampleKeyPairsDataset(self._table, prefix="", start="", end="") expected_key_pairs = [("", "r1"), ("r1", "r3"), ("r3", "")] self.runSampleKeyPairsTest(ds, expected_key_pairs)
def testSampleKeyPairsOffsetRanges(self): ds = bigtable_api._BigtableSampleKeyPairsDataset(self._table, prefix="", start="r2", end="r4") expected_key_pairs = [("r2", "r3"), ("r3", "r4")] self.runSampleKeyPairsTest(ds, expected_key_pairs)
def testSampleKeyPairsPrefixAndEndKey(self): ds = bigtable_api._BigtableSampleKeyPairsDataset( self._table, prefix="r", start="", end="r3") itr = ds.make_initializable_iterator() with self.test_session() as sess: with self.assertRaises(errors.InvalidArgumentError): sess.run(itr.initializer)
def testSampleKeyPairsPrefixAndEndKey(self): ds = bigtable_api._BigtableSampleKeyPairsDataset( self._table, prefix="r", start="", end="r3") itr = ds.make_initializable_iterator() with self.test_session() as sess: with self.assertRaises(errors.InvalidArgumentError): sess.run(itr.initializer)
def testSampleKeyPairEverything(self): ds = bigtable_api._BigtableSampleKeyPairsDataset( self._table, prefix="", start="", end="") expected_key_pairs = [("", "r1"), ("r1", "r3"), ("r3", "")] self.runSampleKeyPairsTest(ds, expected_key_pairs)
def testSampleKeyPairsOffsetRanges(self): ds = bigtable_api._BigtableSampleKeyPairsDataset( self._table, prefix="", start="r2", end="r4") expected_key_pairs = [("r2", "r3"), ("r3", "r4")] self.runSampleKeyPairsTest(ds, expected_key_pairs)