moves testing project into the main repo

Author: lschwetlick

testing_project/.python-version

@@ -0,0 +1 @@
+3.10.6

testing_project/LICENSE.txt

@@ -0,0 +1,6 @@
+The material in this repository is released under the
+CC Attribution-Share Alike 4.0 International
+license.
+
+Full license text available at
+https://creativecommons.org/licenses/by-sa/4.0/

testing_project/README.md

@@ -0,0 +1 @@
+# Testing Project

testing_project/demos/conftest_example.py

@@ -0,0 +1,28 @@
+import numpy as np
+import pytest
+
+
+# add a commandline option to pytest
+def pytest_addoption(parser):
+    """Add random seed option to py.test.
+    """
+    parser.addoption('--seed', dest='seed', type=int, action='store',
+                     help='set random seed')
+
+
+# configure pytest to automatically set the rnd seed if not passed on CLI
+def pytest_configure(config):
+    seed = config.getvalue("seed")
+    # if seed was not set by the user, we set one now
+    if seed is None or seed == ('NO', 'DEFAULT'):
+        config.option.seed = int(np.random.randint(2 ** 31 - 1))
+
+
+def pytest_report_header(config):
+    return f'Using random seed: {config.option.seed}'
+
+
+@pytest.fixture
+def random_state(request):
+    random_state = np.random.RandomState(request.config.option.seed)
+    return random_state

testing_project/demos/test_parametrize.py

@@ -0,0 +1,28 @@
+import pytest
+
+
+def test_for_loop_simple():
+    cases = [1, 2, 3]
+    for a in cases:
+        assert a > 0
+
+
+@pytest.mark.parametrize('a', [1, 2, 3])
+def test_parametrize_simple(a):
+    # This test will be run 3 times, with a=1, a=2, and a=3
+    assert a > 0
+
+
+def test_for_loop_multiple():
+    cases = [(1, 'hi', 'hi'), (2, 'no', 'nono')]
+    for a, b, expected in cases:
+        result = b * a
+        assert result == expected
+
+
+@pytest.mark.parametrize('a, b, expected', [(1, 'hi', 'hi'), (2, 'no', 'nono')])
+def test_parametrize_multiple(a, b, expected):
+    # This test will be run 2 times, with a=1, b='hi', expected='hi'
+    # and a=2, b='no', expected='nono'
+    result = b * a
+    assert result == expected

testing_project/logistic.py

@@ -0,0 +1 @@
+# Your code goes here

testing_project/logistic_fit.py

@@ -0,0 +1,32 @@
+import numpy as np
+
+from logistic import iterate_f
+
+
+def fit_r(xs):
+    """ Takes a population trajectory and returns the value of r that generated it.
+
+    By far not the most efficient method, but it always finds the optimal value of r with 1/1000
+    precision.
+
+    Parameters
+    ----------
+    xs : list of float
+        A population trajectory.
+
+    Returns
+    -------
+    r: float
+        The value of r that generated the population trajectory.
+    """
+    xs = np.asarray(xs)
+    x0 = xs[0]
+    it = len(xs) - 1
+
+    def error(r):
+        return np.linalg.norm(xs - iterate_f(it, x0, r))
+
+    errors = []
+    for r in np.linspace(0, 4, 4001):
+        errors.append((r, error(r)))
+    return min(errors, key=lambda x: x[1])[0]

testing_project/plot_logistic.py

@@ -0,0 +1,70 @@
+"""Usage:
+```
+plot_trajectory(100, 3.6, 0.1)
+plot_bifurcation(2.5, 4.2, 0.001)
+```
+"""
+import numpy as np
+from matplotlib import pyplot as plt
+
+from logistic import iterate_f
+
+
+def plot_trajectory(n, r, x0, fname="single_trajectory.png"):
+    """
+    Saves a plot of a single trajectory of the logistic function
+
+    inputs
+        n: int (number of iterations)
+        r: float (r value for the logistic function)
+        x0: float (between 0 and 1, starting point for the iteration)
+        fname: str (filename to which to save the image)
+
+    returns
+        fig, ax (matplotlib objects)
+    """
+    xs = iterate_f(n, x0, r)
+    fig, ax = plt.subplots(figsize=(10, 5))
+    ax.plot(list(range(n)), xs)
+    fig.suptitle('Logistic Function')
+
+    fig.savefig(fname)
+    return fig, ax
+
+
+def plot_bifurcation(start, end, step, fname="bifurcation.png", it=100000,
+                     last=300):
+    """
+    Saves a plot of the bifurcation diagram of the logistic function. The
+    `start`, `end`, and `step` parameters define for which r values to
+    calculate the logistic function. If you space them too closely, it might
+    take a very long time, if you dont plot enough, your bifurcation diagram
+    won't be informative. Choose wisely!
+
+    inputs
+        start, end, step:  float (which r values to calculate the logistic
+                                  function for)
+        fname: str (filename to which to save the image)
+        it: int (how many iterations to run for each r value)
+        last: int (how many of the last iterates to plot)
+
+
+    returns
+        fig, ax (matplotlib objects)
+    """
+    r_range = np.arange(start, end, step)
+    x = []
+    y = []
+
+    for r in r_range:
+        xs = iterate_f(it, 0.1, r)
+        all_xs = xs[len(xs) - last::].copy()
+        unique_xs = np.unique(all_xs)
+        y.extend(unique_xs)
+        x.extend(np.ones(len(unique_xs)) * r)
+
+    fig, ax = plt.subplots(figsize=(20, 10))
+    ax.scatter(x, y, s=0.1, color='k')
+    ax.set_xlabel("r")
+    fig.savefig(fname)
+    return fig, ax

testing_project/pytest.ini

@@ -0,0 +1,4 @@
+# pytest.ini
+[pytest]
+norecursedirs = *
+

testing_project/solution/conftest.py

@@ -0,0 +1,28 @@
+import numpy as np
+import pytest
+
+
+# add a commandline option to pytest
+def pytest_addoption(parser):
+    """Add random seed option to py.test.
+    """
+    parser.addoption('--seed', dest='seed', type=int, action='store',
+                     help='set random seed')
+
+
+# configure pytest to automatically set the rnd seed if not passed on CLI
+def pytest_configure(config):
+    seed = config.getvalue("seed")
+    # if seed was not set by the user, we set one now
+    if seed is None or seed == ('NO', 'DEFAULT'):
+        config.option.seed = int(np.random.randint(2 ** 31 - 1))
+
+
+def pytest_report_header(config):
+    return f'Using random seed: {config.option.seed}'
+
+
+@pytest.fixture
+def random_state(request):
+    random_state = np.random.RandomState(request.config.option.seed)
+    return random_state