def test_set_reader(self): reader = make_reader(1) splitter = UniformSplitter(self.project_id, self.vault_id, reader) self.assertEqual(reader, splitter.input_stream) new_reader = make_reader(2) splitter.input_stream = new_reader self.assertEqual(new_reader, splitter.input_stream) splitter._set_state('processing') with self.assertRaises(RuntimeError): splitter.input_stream = reader
def test_set_reader_no_read(self): class X(object): def tell(): pass reader = make_reader(1) splitter = UniformSplitter(self.project_id, self.vault_id, reader) with self.assertRaises(AttributeError): splitter.input_stream = X()