From dde2910c597bf1459bda710e17a138455eca8078 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?P=C4=93teris=20Caune?= <cuu508@gmail.com>
Date: Mon, 12 Aug 2019 14:41:50 +0300
Subject: [PATCH] Cleanup.

---
 hc/api/tests/test_notify.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/hc/api/tests/test_notify.py b/hc/api/tests/test_notify.py
index a1ecd31e..ac37e926 100644
--- a/hc/api/tests/test_notify.py
+++ b/hc/api/tests/test_notify.py
@@ -668,10 +668,6 @@ class NotifyTestCase(BaseTestCase):
     def test_not_implimented(self):
         self._setup_data("webhook", "http://example")
         self.channel.kind = "invalid"
-        try:
+
+        with self.assertRaises(NotImplementedError):
             self.channel.notify(self.check)
-            # Code should not reach here
-            assert False
-        except NotImplementedError:
-            # We expect to be here
-            assert True