bites 148 - grouping

Author: [email protected]

.gitignore

@@ -67,3 +67,5 @@
 /140-pandas/README.md
 /118/README.md
 /116/README.md
+/187/README.md
+/148/README.md

148/grouping.py

@@ -0,0 +1,34 @@
+from itertools import groupby
+
+cars = [
+    # need mock data? -> https://www.mockaroo.com == awesome
+    ('Mercedes-Benz', '300D'), ('Mercedes-Benz', '600SEL'),
+    ('Toyota', 'Avalon'), ('Ford', 'Bronco'),
+    ('Chevrolet', 'Cavalier'), ('Chevrolet', 'Corvette'),
+    ('Mercedes-Benz', 'E-Class'), ('Hyundai', 'Elantra'),
+    ('Volkswagen', 'GTI'), ('Toyota', 'Highlander'),
+    ('Chevrolet', 'Impala'), ('Nissan', 'Maxima'),
+    ('Ford', 'Mustang'), ('Kia', 'Optima'),
+    ('Volkswagen', 'Passat'), ('Nissan', 'Pathfinder'),
+    ('Volkswagen', 'Routan'), ('Hyundai', 'Sonata'),
+    ('Kia', 'Sorento'), ('Kia', 'Sportage'),
+    ('Ford', 'Taurus'), ('Nissan', 'Titan'),
+    ('Toyota', 'Tundra'), ('Hyundai', 'Veracruz'),
+]
+
+
+def group_cars_by_manufacturer(cars):
+    """Iterate though the list of (manufacturer, model) tuples
+       of the cars list defined above and generate the output as described
+       in the Bite description (see the tests for the full output).
+       
+       No return here, just print to the console. We use pytest > capfd to
+       validate your output :)
+    """
+    cars = sorted(cars, key=lambda x: x[0])
+    for key, group in groupby(cars, lambda x: x[0]):
+        cars_text = "\n- ".join(list(map(lambda x: x[1], group)))
+        print(key.upper() + "\n- " + cars_text + "\n")
+
+
+# group_cars_by_manufacturer(cars)

148/test_grouping.py

@@ -0,0 +1,49 @@
+from grouping import cars, group_cars_by_manufacturer
+
+expected_output = """
+CHEVROLET
+- Cavalier
+- Corvette
+- Impala
+
+FORD
+- Bronco
+- Mustang
+- Taurus
+
+HYUNDAI
+- Elantra
+- Sonata
+- Veracruz
+
+KIA
+- Optima
+- Sorento
+- Sportage
+
+MERCEDES-BENZ
+- 300D
+- 600SEL
+- E-Class
+
+NISSAN
+- Maxima
+- Pathfinder
+- Titan
+
+TOYOTA
+- Avalon
+- Highlander
+- Tundra
+
+VOLKSWAGEN
+- GTI
+- Passat
+- Routan
+"""
+
+
+def test_group_cars_by_manufacturer(capfd):
+    group_cars_by_manufacturer(cars)
+    actual_output, _ = capfd.readouterr()
+    assert actual_output.strip() == expected_output.strip()