Add find_items method and initial test.

Author: Tomasz-Kluczkowski

data_extractor.py

@@ -0,0 +1,21 @@
+from websites.resources.data import WEBSITES
+
+
+class DataExtractor:
+    """
+    Use to extract, cleanse and amend incorrect website data collection.
+    """
+    def __init__(self):
+        self.data = WEBSITES
+
+    def find_items(self, value):
+        """
+        Find and return a new list of items where key "value" is greater than or equal to parameter value.
+        :return: list(dict), list of dictionaries matching the above filtering rule.
+        """
+        return [item for item in self.data if item.get('value') and item.get('value') >= value]
+
+
+# data_extractor = DataExtractor()
+# print(data_extractor.find_items(4))
+# print(len(data_extractor.find_items(4)))

tests/__init__.py

@@ -0,0 +1,32 @@
+import pytest
+
+from data_extractor import DataExtractor
+
+data_extractor = DataExtractor()
+
+
+class TestDataExtractor:
+
+    def test_find_items(self):
+        expected = [
+            {
+                'name': 'Google',
+                'url': 'https://www.google.co.uk',
+                'domain': 'google.co.uk',
+                'secure': True,
+                'value': 5},
+            {
+                'name': 'Facebook',
+                'url': 'https://developers.facebook.com/blog/post/2018/10/02/facebook-login-update/',
+                'domain': 'facebook.com',
+                'secure': True,
+                'value': 4},
+            {
+                'name': 'YouTube',
+                'url': 'https://www.youtube.com/watch?v=09Cd7NKKvDc',
+                'domain': 'youtube.com',
+                'secure': True,
+                'value': 5
+            }
+        ]
+        assert data_extractor.find_items(4) == expected