python/nested-lookup
1
0
Fork 0
Code Issues 6 Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #8 from rameshrvr/master

Browse source 
Fix for issue #7 (Enable key lookup after the key is found)
This commit is contained in:
Russell Ballestrini 2018-10-16 07:54:55 -04:00 • committed by GitHub
commit 08967c817b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 70 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
README.rst
View file

@ -1,7 +1,7 @@
nested_lookup nested_lookup
############# #############
.. image:: https://img.shields.io/badge/pypi-0.1.5-green.svg .. image:: https://img.shields.io/badge/pypi-0.1.7-green.svg
:target: https://pypi.python.org/pypi/nested-lookup :target: https://pypi.python.org/pypi/nested-lookup
A small Python library which enables: A small Python library which enables:

2
nested_lookup/nested_lookup.py
View file

@ -27,7 +27,7 @@ def _nested_lookup(key, document, wild=False, with_keys=False):
yield k, v yield k, v
else: else:
yield v yield v
elif isinstance(v, dict): if isinstance(v, dict):
for result in _nested_lookup(key, v, wild=wild, with_keys=with_keys): for result in _nested_lookup(key, v, wild=wild, with_keys=with_keys):
yield result yield result
elif isinstance(v, list): elif isinstance(v, list):

74
test_nested_loopkup.py
View file

@ -17,6 +17,27 @@ class TestNestedLookup(TestCase):
"email_address": "test4@example.com", "email_address": "test4@example.com",
}, },
} }
self.subject_dict3 = {
"build_version": {
"model_name": "MacBook Pro",
"build_version": {
"processor_name": "Intel Core i7",
"processor_speed": "2.7 GHz",
"core_details": {
"build_version": "4",
"l2_cache(per_core)": "256 KB"
}
},
"number_of_cores": "4",
"memory": "256 KB",
},
"os_details": {
"product_version": "10.13.6",
"build_version": "17G65"
},
"name": "Test",
"date": "YYYY-MM-DD HH:MM:SS"
}
def test_nested_lookup(self): def test_nested_lookup(self):
results = nested_lookup("d", self.subject_dict) results = nested_lookup("d", self.subject_dict)
@ -47,7 +68,8 @@ class TestNestedLookup(TestCase):
self.assertSetEqual({100, 200}, set(results)) self.assertSetEqual({100, 200}, set(results))
def test_wild_nested_lookup(self): def test_wild_nested_lookup(self):
results = nested_lookup(key="mail", document=self.subject_dict2, wild=True) results = nested_lookup(
key="mail", document=self.subject_dict2, wild=True)
self.assertEqual(4, len(results)) self.assertEqual(4, len(results))
self.assertIn("test1@example.com", results) self.assertIn("test1@example.com", results)
self.assertIn("test2@example.com", results) self.assertIn("test2@example.com", results)
@ -61,8 +83,9 @@ class TestNestedLookup(TestCase):
self.assertIn("email_address", matches) self.assertIn("email_address", matches)
self.assertIn("secondary_email", matches) self.assertIn("secondary_email", matches)
self.assertIn("EMAIL_RECOVERY", matches) self.assertIn("EMAIL_RECOVERY", matches)
self.assertSetEqual( self.assertSetEqual({
{"test1@example.com", "test4@example.com"}, set(matches["email_address"]) "test1@example.com", "test4@example.com"},
set(matches["email_address"])
) )
self.assertIn("test2@example.com", matches["secondary_email"]) self.assertIn("test2@example.com", matches["secondary_email"])
@ -72,6 +95,37 @@ class TestNestedLookup(TestCase):
self.assertEqual(2, len(matches["d"])) self.assertEqual(2, len(matches["d"]))
self.assertSetEqual({100, 200}, set(matches["d"])) self.assertSetEqual({100, 200}, set(matches["d"]))
def test_after_key_is_found(self):
result = nested_lookup(
key='build_version', document=self.subject_dict3
)
self.assertEqual(4, len(result))
self.assertIn('4', result)
self.assertIn('17G65', result)
match1 = {
'processor_name': 'Intel Core i7',
'processor_speed': '2.7 GHz',
'core_details': {
'build_version': '4',
'l2_cache(per_core)': '256 KB'
}
}
self.assertIn(match1, result)
match2 = {
'build_version': {
'processor_name': 'Intel Core i7',
'processor_speed': '2.7 GHz',
'core_details': {
'build_version': '4',
'l2_cache(per_core)': '256 KB'
}
},
'memory': '256 KB',
'model_name': 'MacBook Pro',
'number_of_cores': '4'
}
self.assertIn(match2, result)
class TestGetAllKeys(TestCase): class TestGetAllKeys(TestCase):
@ -90,7 +144,9 @@ class TestGetAllKeys(TestCase):
"total_number_of_cores": "4", "total_number_of_cores": "4",
"memory": "16 GB", "memory": "16 GB",
}, },
"os_details": {"product_version": "10.13.6", "build_version": "17G65"}, "os_details": {
"product_version": "10.13.6", "build_version": "17G65"
},
"name": "Test", "name": "Test",
"date": "YYYY-MM-DD HH:MM:SS", "date": "YYYY-MM-DD HH:MM:SS",
} }
@ -115,8 +171,14 @@ class TestGetAllKeys(TestCase):
"hardware_details": { "hardware_details": {
"model_name": "MacBook Pro", "model_name": "MacBook Pro",
"processor_details": [ "processor_details": [
{"processor_name": "Intel Core i7", "processor_speed": "2.7 GHz"}, {
{"total_numberof_cores": "4", "l2_cache(per_core)": "256 KB"}, "processor_name": "Intel Core i7",
"processor_speed": "2.7 GHz"
},
{
"total_numberof_cores": "4",
"l2_cache(per_core)": "256 KB"
},
], ],
"total_number_of_cores": "4", "total_number_of_cores": "4",
"memory": "16 GB", "memory": "16 GB",