diff --git a/test/test_devalue.py b/test/test_devalue.py index 4ae1c92f25..1130f1f94b 100644 --- a/test/test_devalue.py +++ b/test/test_devalue.py @@ -8,6 +8,7 @@ import datetime as dt +import json import math import re import unittest @@ -222,8 +223,12 @@ def test_devalue_parse_cyclical(self): def test_devalue_parse_revivers(self): self.assertEqual( - devalue.parse([['Custom', 1], {'a': 2}, 'b'], revivers={'Custom': lambda x: x}), - {'a': 'b'}, 'revivers') + devalue.parse([['indirect', 1], {'a': 2}, 'b'], revivers={'indirect': lambda x: x}), + {'a': 'b'}, 'revivers (indirect)') + + self.assertEqual( + devalue.parse([['parse', 1], '{"a":0}'], revivers={'parse': lambda x: json.loads(x)}), + {'a': 0}, 'revivers (parse)') if __name__ == '__main__':