bites 140

Author: [email protected]

.gitignore

@@ -64,3 +64,4 @@
 /Pipfile.lock
 /146/README.md
 /140/README.md
+/140-pandas/README.md

140-pandas/medals.py

@@ -0,0 +1,11 @@
+import pandas as pd
+
+data = "https://bites-data.s3.us-east-2.amazonaws.com/summer.csv"
+
+
+def athletes_most_medals(data=data):
+    df = pd.read_csv(data)
+    # x = data.head(5)
+    Men_most = df[(df.Gender == "Men")]["Athlete"].value_counts()
+    Women_most = df[(df.Gender == "Women")]["Athlete"].value_counts()
+    return {Men_most.index[0]: Men_most.values[0], Women_most.index[0]: Women_most.values[0]}

140-pandas/test_medals.py

@@ -0,0 +1,17 @@
+from medals import data, athletes_most_medals
+
+
+def test_athletes_most_medals_default_csv():
+    ret = athletes_most_medals()
+    assert len(ret) == 2
+    assert ret["LATYNINA, Larisa"] == 18
+    assert ret["PHELPS, Michael"] == 22
+
+
+def test_smaller_csv_and_guarantee_checking_male_and_female():
+    ret = athletes_most_medals(
+        data.replace('summer', 'summer_2008-2012')
+    )
+    assert len(ret) == 2
+    assert ret["PHELPS, Michael"] == 14
+    assert ret["COUGHLIN, Natalie"] == 7  # not LOCHTE, Ryan