def testInsertGetAndRead(self): records = RecordIORecordsZipped() self.insertABC(records) self.assertEqual(len(records), 3) self.assertEqual({ "a": ("aa", ), "b": ("bb", ), "c": ("cc", ) }, self.getResult(records)) records = RecordIORecordsZipped(records.get_data()) records.insert(("b", "new")) self.assertEqual(len(records), 3) self.assertEqual(records["b"], ("b", "new")) self.assertTrue("a" in records) self.assertFalse("z" in records) records = RecordIORecordsZipped(records.get_data()) records.insert(("b", "bb")) self.assertEqual({ "a": ("aa", ), "b": ("bb", ), "c": ("cc", ) }, self.getResult(records.read())) self.assertEqual({ "a": ("aa", ), "b": ("bb", ), "c": ("cc", ) }, self.getResult(records.read(("", ), ("d", )))) self.assertEqual({"b": ("bb", )}, self.getResult(records.read(("b", ), ("c", ))))
def testInsertGetAndRead(self): records = RecordIORecordsZipped() self.insertABC(records) self.assertEqual(len(records), 3) self.assertEqual({"a": ("aa",), "b": ("bb",), "c": ("cc",)}, self.getResult(records)) records = RecordIORecordsZipped(records.get_data()) records.insert(("b", "new")) self.assertEqual(len(records), 3) self.assertEqual(records["b"], ("b", "new")) self.assertTrue("a" in records) self.assertFalse("z" in records) records = RecordIORecordsZipped(records.get_data()) records.insert(("b", "bb")) self.assertEqual({"a": ("aa",), "b": ("bb",), "c": ("cc",)}, self.getResult(records.read())) self.assertEqual({"a": ("aa",), "b": ("bb",), "c": ("cc",)}, self.getResult(records.read(("",), ("d",)))) self.assertEqual({"b": ("bb",)}, self.getResult(records.read(("b",), ("c",))))
def testSplit(self): records = RecordIORecordsZipped() self.insertABC(records) records.insert(("d", test_helper.uncompressableString(ZIP_CHUNKS / 2))) records.insert(("e", test_helper.uncompressableString(ZIP_CHUNKS))) lo, hi, middle = records.split() lo = RecordIORecordsZipped(lo) hi = RecordIORecordsZipped(hi) self.assertEqual(middle[0], "e") self.assertEqual(["a", "b", "c", "d"], list(sorted(self.getResult(lo).keys()))) self.assertEqual({"a": ("aa",), "b": ("bb",), "c": ("cc",)}, self.getResult(lo.read(("",), ("d",)))) self.assertEqual(["e"], self.getResult(hi).keys())
def testSplit(self): records = RecordIORecordsZipped() self.insertABC(records) records.insert(("d", test_helper.uncompressableString(ZIP_CHUNKS / 2))) records.insert(("e", test_helper.uncompressableString(ZIP_CHUNKS))) lo, hi, middle = records.split() lo = RecordIORecordsZipped(lo) hi = RecordIORecordsZipped(hi) self.assertEqual(middle[0], "e") self.assertEqual(["a", "b", "c", "d"], list(sorted(self.getResult(lo).keys()))) self.assertEqual({ "a": ("aa", ), "b": ("bb", ), "c": ("cc", ) }, self.getResult(lo.read(("", ), ("d", )))) self.assertEqual(["e"], self.getResult(hi).keys())