def test_read_empty_data(self): data = [] gen = GeneratorIO(iter(data)) self.assertEqual(gen.read(10), "") data = ["", "", ""] gen = GeneratorIO(iter(data)) self.assertEqual(gen.read(10), "")
def test_read_1_by_1_byte(self): data = ["a", "bc", "", "d"] gen = GeneratorIO(iter(data)) self.assertEqual(gen.read(1), "a") self.assertEqual(gen.read(1), "b") self.assertEqual(gen.read(1), "c") self.assertEqual(gen.read(1), "d") self.assertEqual(gen.read(1), "")
def test_read_more_than_data_size(self): data = ["a", "bc", "", "d"] gen = GeneratorIO(iter(data)) self.assertEqual(gen.read(10), "abcd") self.assertEqual(gen.read(10), "")