OR updates

Author: iansedano

python-first/README.md

@@ -1,3 +1,5 @@
 # How to Find the First Match from an Iterable in Python
 
 The code samples and supporting materials for the corresponding tutorial on Real Python.
+
+Country data obtained from https://github.com/samayo/country-json

python-first/chart.py

@@ -2,20 +2,38 @@
 
 import matplotlib.pyplot as plt
 
-from test_cases import build_list
+from test_fixtures import build_list
 
 TIMEIT_TIMES = 100_000
 LIST_SIZE = 500
 POSITION_INCREMENT = 10
 
 looping_times = []
 generator_times = []
+in_times = []
 positions = []
 
+
+def find_match_loop(list_to_search, item_to_find):
+    for val in list_to_search:
+        if val == item_to_find:
+            return val
+    return None
+
+
+def find_match_gen(list_to_search, item_to_find):
+    return next((val for val in list_to_search if val == item_to_find), None)
+
+
+def find_match_in(list_to_search, item_to_find):
+    if item_to_find in list_to_search:
+        return item_to_find
+
+
 for position in range(0, LIST_SIZE, POSITION_INCREMENT):
     print(
         f"Progress {position/LIST_SIZE:.0%}",
-        end=f"{3 * ' '}\r",  # Clear previous characters and reset cursor
+        end=f"{3 * ' '}\r",  # Clear previous characters and resets cursor
     )
 
     positions.append(position)
@@ -26,16 +44,21 @@
 
     looping_times.append(
         timeit(
-            "find_match_loop(list_to_search, 'Real Python', None)",
-            setup="from looping import find_match_loop",
+            "find_match_loop(list_to_search, 'Real Python')",
             globals=globals(),
             number=TIMEIT_TIMES,
         )
     )
     generator_times.append(
         timeit(
-            "find_match_gen(list_to_search, 'Real Python', None)",
-            setup="from generator import find_match_gen",
+            "find_match_gen(list_to_search, 'Real Python')",
+            globals=globals(),
+            number=TIMEIT_TIMES,
+        )
+    )
+    in_times.append(
+        timeit(
+            "find_match_in(list_to_search, 'Real Python')",
             globals=globals(),
             number=TIMEIT_TIMES,
         )
@@ -47,6 +70,7 @@
 
 plot = ax.plot(positions, looping_times, label="loop")
 plot = ax.plot(positions, generator_times, label="generator")
+plot = ax.plot(positions, in_times, label="in")
 
 plt.xlim([0, LIST_SIZE])
 plt.xlabel("Position of element to be found")