I think the `with self.assertRaises` variant reads better, but I see that you just continued using the existing style here. Up to you if you want to fix it now.