def test_set_invalid_state(self):
reader = make_reader(1)
splitter = UniformSplitter(self.project_id,
self.vault_id,
reader)
with self.assertRaises(errors.ParameterConstraintError):
splitter._set_state('bad state')
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
