def test_format_results(self): """Test format_results.""" test_dict = { "nested": { "bool": True, "nested_key": "nested_value" }, "test_key": "test_value", } mute_map = MutableMap(**test_dict.copy()) assert LookupHandler.format_results(test_dict) == test_dict assert LookupHandler.format_results(mute_map) == test_dict assert (LookupHandler.format_results( test_dict, get="test_key") == test_dict["test_key"]) assert (LookupHandler.format_results( mute_map, get="test_key") == mute_map.test_key) assert (LookupHandler.format_results( mute_map, get="nested") == mute_map.nested.data) assert (LookupHandler.format_results( mute_map, get="nested.nested_key") == mute_map.nested.nested_key) assert LookupHandler.format_results(mute_map, get="nested.bool") assert LookupHandler.format_results(mute_map, transform="str") == json.dumps( json.dumps(test_dict, indent=0)) assert LookupHandler.format_results(mute_map, transform="str", indent=2) == json.dumps( json.dumps(test_dict, indent=2)) assert (LookupHandler.format_results(mute_map, get="nested.bool", transform="str") == '"True"') with pytest.raises(TypeError): LookupHandler.format_results(["something"], get="key")
def test_format_results(self): """Test format_results.""" test_dict = { 'nested': { 'bool': True, 'nested_key': 'nested_value' }, 'test_key': 'test_value' } mute_map = MutableMap(**test_dict.copy()) assert LookupHandler.format_results(test_dict) == test_dict assert LookupHandler.format_results(mute_map) == test_dict assert LookupHandler.format_results( test_dict, get='test_key') == test_dict['test_key'] assert LookupHandler.format_results( mute_map, get='test_key') == mute_map.test_key assert LookupHandler.format_results( mute_map, get='nested') == mute_map.nested.data assert LookupHandler.format_results( mute_map, get='nested.nested_key') == mute_map.nested.nested_key assert LookupHandler.format_results(mute_map, get='nested.bool') assert LookupHandler.format_results(mute_map, transform='str') == json.dumps( json.dumps(test_dict, indent=0)) assert LookupHandler.format_results(mute_map, transform='str', indent=2) == json.dumps( json.dumps(test_dict, indent=2)) assert LookupHandler.format_results(mute_map, get='nested.bool', transform='str') == '"True"' with self.assertRaises(TypeError): LookupHandler.format_results(['something'], get='key')