def test_raise_on_undef(self): """Test the static class RaiseOnUndef.""" self.assertRaises(AssertionError, lambda: RaiseOnUndef()) self.assertEqual(RaiseOnUndef.get_level(), 0) RaiseOnUndef.set_level(2) self.assertEqual(RaiseOnUndef.get_level(), 2) RaiseOnUndef.set_level(4) self.assertEqual(RaiseOnUndef.get_level(), 4) RaiseOnUndef.reset() self.assertEqual(RaiseOnUndef.get_level(), 0)
def test_make_or_raise_undef(self): """Test make_or_raise_undef() together with RaiseOnUndef.""" self.assertEqual(RaiseOnUndef.get_level(), 0) self.assertIs(make_or_raise_undef(), Undef()) RaiseOnUndef.set_level(1) self.assertRaises(UndefException, lambda: make_or_raise_undef()) self.assertIs(make_or_raise_undef(2), Undef()) RaiseOnUndef.set_level(2) self.assertRaises(UndefException, lambda: make_or_raise_undef(2)) RaiseOnUndef.reset() self.assertIs(make_or_raise_undef(2), Undef())
def test_make_or_raise_undef(self): """Test make_or_raise_undef() together with RaiseOnUndef.""" try: self.assertEqual(RaiseOnUndef.get_level(), 0) self.assertIs(make_or_raise_undef(), Undef()) RaiseOnUndef.set_level(1) self.assertRaises(UndefException, lambda: make_or_raise_undef()) self.assertIs(make_or_raise_undef(2), Undef()) RaiseOnUndef.set_level(2) self.assertRaises(UndefException, lambda: make_or_raise_undef(2)) RaiseOnUndef.reset() self.assertIs(make_or_raise_undef(2), Undef()) except: # Make sure RaiseOnUndef level gets reset. RaiseOnUndef.reset() raise