Update Context test to expect KeyError instead of AttributeError

The WorkChain Context is now an aiida.common.extenddicts.AttributeDict which should raise a KeyError when ['key'] referencing excepts
aiidateam · Jan 17, 2018 · a140af9 · a140af9
1 parent a2c9ce9
commit a140af9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aiida/backends/tests/work/work_chain.py b/aiida/backends/tests/work/work_chain.py
@@ -114,7 +114,7 @@ def test_dict(self):
         self.assertEqual(wc.ctx['new_attr'], 5)
 
         del wc.ctx['new_attr']
-        with self.assertRaises(AttributeError):
+        with self.assertRaises(KeyError):
             wc.ctx['new_attr']