tests!

Author: billillib

.gitignore

@@ -1,3 +1,5 @@
+*.pytest_cache/
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

days/07-09-data-structures/code/Day3.ipynb

@@ -2,27 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 48,
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "N/A\n"
-     ]
-    }
-   ],
-   "source": [
-    "if 'Zebra' in us_state_abbrev.keys():\n",
-    "    print(us_state_abbrev['Zebra'])\n",
-    "else:\n",
-    "    print(NOT_FOUND)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 125,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -110,7 +90,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 11,
    "metadata": {},
    "outputs": [],
    "source": [

days/07-09-data-structures/code/challenge.py

@@ -0,0 +1,79 @@
+us_state_abbrev = {'Alabama': 'AL', 'Alaska': 'AK', 'Arizona': 'AZ',
+                   'Arkansas': 'AR', 'California': 'CA', 'Colorado': 'CO',
+                   'Connecticut': 'CT', 'Delaware': 'DE', 'Florida': 'FL',
+                   'Georgia': 'GA', 'Hawaii': 'HI', 'Idaho': 'ID',
+                   'Illinois': 'IL', 'Indiana': 'IN', 'Iowa': 'IA',
+                   'Kansas': 'KS', 'Kentucky': 'KY', 'Louisiana': 'LA',
+                   'Maine': 'ME', 'Maryland': 'MD', 'Massachusetts': 'MA',
+                   'Michigan': 'MI', 'Minnesota': 'MN', 'Mississippi': 'MS',
+                   'Missouri': 'MO', 'Montana': 'MT', 'Nebraska': 'NE',
+                   'Nevada': 'NV', 'New Hampshire': 'NH', 'New Jersey': 'NJ',
+                   'New Mexico': 'NM', 'New York': 'NY',
+                   'North Carolina': 'NC', 'North Dakota': 'ND',
+                   'Ohio': 'OH', 'Oklahoma': 'OK', 'Oregon': 'OR',
+                   'Pennsylvania': 'PA', 'Rhode Island': 'RI',
+                   'South Carolina': 'SC', 'South Dakota': 'SD',
+                   'Tennessee': 'TN', 'Texas': 'TX', 'Utah': 'UT',
+                   'Vermont': 'VT', 'Virginia': 'VA', 'Washington': 'WA',
+                   'West Virginia': 'WV', 'Wisconsin': 'WI', 'Wyoming': 'WY'}
+
+states = ['Oklahoma', 'Kansas', 'North Carolina', 'Georgia', 'Oregon',
+          'Mississippi', 'Minnesota', 'Colorado', 'Alabama',
+          'Massachusetts', 'Arizona', 'Connecticut', 'Montana',
+          'West Virginia', 'Nebraska', 'New York', 'Nevada', 'Idaho',
+          'New Jersey', 'Missouri', 'South Carolina', 'Pennsylvania',
+          'Rhode Island', 'New Mexico', 'Alaska', 'New Hampshire',
+          'Tennessee', 'Washington', 'Indiana', 'Hawaii', 'Kentucky',
+          'Virginia', 'Ohio', 'Wisconsin', 'Maryland', 'Florida',
+          'Utah', 'Maine', 'California', 'Vermont', 'Arkansas', 'Wyoming',
+          'Louisiana', 'North Dakota', 'South Dakota', 'Texas',
+          'Illinois', 'Iowa', 'Michigan', 'Delaware']
+
+NOT_FOUND = 'N/A'
+
+
+def get_every_nth_state(states=states, n=10):
+    """Return a list with every nth item (default argument n=10, so every
+       10th item) of the states list above (remember: lists keep order)"""
+    state_list = [i for idx, i in enumerate(states) if (idx + 1) % n == 0]
+
+    return state_list
+
+
+def get_state_abbrev(state_name, us_state_abbrev=us_state_abbrev):
+    """Look up a state abbreviation by querying the us_state_abbrev
+       dict by full state name, for instance 'Alabama' returns 'AL',
+       'Illinois' returns 'IL'.
+       If the state is not in the dict, return 'N/A' which we stored
+       in the NOT_FOUND constant (takeaway: dicts are great for lookups)"""
+    # if state_name in us_state_abbrev.keys():
+    #     return us_state_abbrev[state_name]
+    # else:
+    #     return NOT_FOUND
+    return us_state_abbrev.get(state_name) or NOT_FOUND
+
+
+def get_longest_state(data):
+    """Receives data, which can be the us_state_abbrev dict or the states
+       list (see above). It returns the longest state measured by the length
+       of the string"""
+    return max(data, key=len)
+
+
+def combine_state_names_and_abbreviations(us_state_abbrev=us_state_abbrev,
+                                          states=states, n=10):
+    """Get the first 10 state abbreviations ('AL', 'AK', 'AZ', ...) from
+       the us_state_abbrev dict, and the last 10 states from the states
+       list (see above) and combine them into a new list without losing
+       alphabetical order"""
+    # from dict
+    list_a = [v for v in us_state_abbrev.values()]
+    list_a = list_a[0:n]
+    list_a.sort()
+
+    # from list
+
+    states.sort()  # check to see if it only does it within the function scope
+    list_b = states[-n:]
+
+    return list_a + list_b

days/07-09-data-structures/code/challenge_test.py

@@ -0,0 +1,36 @@
+from challenge import (states, get_every_nth_state,
+                       get_longest_state,
+                       combine_state_names_and_abbreviations,
+                       get_state_abbrev)
+
+
+def test_get_every_nth_state():
+
+    test_states = ['a', 'b', 'c', 'd', 'e', 'f']
+
+    assert get_every_nth_state(states=test_states, n=3) == ['c', 'f']
+
+
+def test_get_state_abbrev():
+
+    assert get_state_abbrev(state_name='Alabama') == 'AL'
+
+    assert get_state_abbrev(state_name='Warrshington') == 'N/A'
+
+
+def test_get_longest_state():
+
+    data = {'Washington': 'WA', 'ThisIsAReallyLongStateName': 'ZZ'}
+
+    assert get_longest_state(data=data) == 'ThisIsAReallyLongStateName'
+
+
+def test_combine_state_names_and_abbreviations():
+
+    test_dict = {'c': 100, 'b': 2}
+    test_list = ['z', 'y']
+    test_result = [2, 100, 'y', 'z']
+
+    assert combine_state_names_and_abbreviations(us_state_abbrev=test_dict,
+                                                 states=test_list,
+                                                 n=2) == test_result

days/10-12-pytest/code/hello.py

@@ -0,0 +1,2 @@
+def hello_name(name):
+    return f'hello {name}'

days/10-12-pytest/code/hello_test.py

@@ -0,0 +1,5 @@
+from hello import hello_name
+
+
+def test_hello_name():
+    assert hello_name('bill') == 'hello bill'