(Python) How to test pipeline that raises an exception?

[Jaehyeon Kim]

Hello,

I have a unit testing case and the pipeline raises TypeError. How is it
possible to test? (I don't find unittest assertRaises method in the beam
testing util)

Cheers,
Jaehyeon

    def test_write_to_firehose_with_unsupported_types(self):
        with TestPipeline(options=self.pipeline_opts) as p:
            output = (
                p
                | beam.Create(["one", "two", "three", "four"])
                | "WriteToFirehose" >> WriteToFirehose(self.
delivery_stream_name, True)
                | "CollectResponse" >> beam.Map(lambda e: e["FailedPutCount"
])
            )