NAME = "Jane Doe"
COLLABORATORS = "n/a"

"""
Explanation: Example Assignment

<a href="#Problem-1">Problem 1</a>
<a href="#Problem-2">Problem 2</a>
<a href="#Part-A">Part A</a>
<a href="#Part-B">Part B</a>
<a href="#Part-C">Part C</a>




Before you turn this problem in, make sure everything runs as expected. First, restart the kernel (in the menubar, select Kernel$\rightarrow$Restart) and then run all cells (in the menubar, select Cell$\rightarrow$Run All).
Make sure you fill in any place that says YOUR CODE HERE or "YOUR ANSWER HERE", as well as your name and collaborators below:
End of explanation
"""


# import plotting libraries
%matplotlib inline
import matplotlib.pyplot as plt

"""
Explanation: 
End of explanation
"""


def squares(n):
    """Compute the squares of numbers from 1 to n, such that the 
    ith element of the returned list equals i^2.
    
    """
    return [i**2 for i in xrange(1, n + 1)]

"""
Explanation: Problem 1
Write a function that returns a list of numbers, such that $x_i=i^2$, for $1\leq i \leq n$. Make sure it handles the case where $n<1$ by raising a ValueError.
End of explanation
"""


squares(10)

"""Check that squares returns the correct output for several inputs"""
from nbgrader.tests import assert_equal
assert_equal(squares(1), [1])
assert_equal(squares(2), [1, 4])
assert_equal(squares(10), [1, 4, 9, 16, 25, 36, 49, 64, 81, 100])
assert_equal(squares(11), [1, 4, 9, 16, 25, 36, 49, 64, 81, 100, 121])

"""Check that squares raises an error for invalid inputs"""
from nbgrader.tests import assert_raises
assert_raises(ValueError, squares, 0)
assert_raises(ValueError, squares, -4)

"""
Explanation: Your function should print [1, 4, 9, 16, 25, 36, 49, 64, 81, 100] for $n=10$. Check that it does:
End of explanation
"""


def sum_of_squares(n):
    """Compute the sum of the squares of numbers from 1 to n."""
    return sum(squares(n))

"""
Explanation: Problem 2
Part A
Using your squares function, write a function that computes the sum of the squares of the numbers from 1 to $n$. Your function should call the squares function -- it should NOT reimplement its functionality.
End of explanation
"""


sum_of_squares(10)

"""Check that sum_of_squares returns the correct answer for various inputs."""
assert_equal(sum_of_squares(1), 1)
assert_equal(sum_of_squares(2), 5)
assert_equal(sum_of_squares(10), 385)
assert_equal(sum_of_squares(11), 506)

"""Check that sum_of_squares relies on squares."""
orig_squares = squares
del squares
try:
    assert_raises(NameError, sum_of_squares, 1)
except AssertionError:
    raise AssertionError("sum_of_squares does not use squares")
finally:
    squares = orig_squares

"""
Explanation: The sum of squares from 1 to 10 should be 385. Verify that this is the answer you get:
End of explanation
"""


fig, ax = plt.subplots() # do not delete this line!
x = range(1, 16)
y = [sum_of_squares(x[i]) for i in xrange(len(x))]
ax.plot(x, y)
ax.set_title("Sum of squares from 1 to $n$")
ax.set_xlabel("$n$")
ax.set_ylabel("sum")

"""Check that the axis limits are correct."""
assert_equal(ax.get_xlim(), (1.0, 15.0))

"""Check that the xlabel is set."""
from nbgrader.tests import assert_unequal
assert_unequal(ax.get_xlabel(), "", "xlabel not set")

"""Check that the ylabel is set."""
assert_unequal(ax.get_ylabel(), "", "ylabel not set")

"""Check that the title is set."""
assert_unequal(ax.get_title(), "", "title not set")

"""Check that the correct xdata was used."""
from nbgrader.tests import assert_allclose, assert_same_shape

lines = ax.get_lines()
assert_equal(len(lines), 1)

xdata = lines[0].get_xdata()
xdata_correct = np.arange(1, 16)
assert_same_shape(xdata, xdata_correct)
assert_allclose(xdata, xdata_correct)

"""Check that the correct ydata was used."""
lines = ax.get_lines()
assert_equal(len(lines), 1)

xdata = lines[0].get_xdata()
ydata = lines[0].get_ydata()
ydata_correct = [sum_of_squares(x) for x in xdata]
assert_same_shape(ydata, ydata_correct)
assert_allclose(ydata, ydata_correct)

"""
Explanation: Part B
Using LaTeX math notation, write out the equation that is implemented by your sum_of_squares function.
$\sum_{i=1}^n i^2$
Part C
Create a plot of the sum of squares for $n=1$ to $n=15$. Make sure to appropriately label the $x$-axis and $y$-axis, and to give the plot a title. Set the $x$-axis limits to be 1 (minimum) and 15 (maximum).
End of explanation
"""


