Merge pull request #49 from kylebarron/master

Fix custom test example in README
jupyter · Nov 24, 2021 · 71a350b · 71a350b
2 parents a152137 + 484bac6
commit 71a350b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/README.rst b/README.rst
@@ -73,10 +73,11 @@ Example
         # in the method name, to avoid clashing with any tests that
         # jupyter_kernel_test adds in the future.
         def test_mykernel_stderr(self):
+            self.flush_channels()
             reply, output_msgs = self.execute_helper(code='print_err "oops"')
-            self.assertEqual(output_msgs[0].header['msg_type'], 'stream')
-            self.assertEqual(output_msgs[0].content['name'], 'stderr')
-            self.assertEqual(output_msgs[0].content['text'], 'oops\n')
+            self.assertEqual(output_msgs[0]['header']['msg_type'], 'stream')
+            self.assertEqual(output_msgs[0]['content']['name'], 'stderr')
+            self.assertEqual(output_msgs[0]['content']['text'], 'oops\n')
 
     if __name__ == '__main__':
         unittest.main()