diff --git a/test/test_example_module.py b/test/test_example_module.py index 2de29d6..21cab4e 100644 --- a/test/test_example_module.py +++ b/test/test_example_module.py @@ -1,11 +1,16 @@ import numpy as np -from py_template.example_module import fibonacci +from py_template.example_module import fibonacci, SoftwareGroup -def test_fibonacci(): - """Check the Fibonacci sequence function""" +def test_example_function(): + """Check the Fibonacci sequence function returns correct result""" fib = fibonacci(10) np.testing.assert_allclose(actual=fib, desired=[1, 1, 2, 3, 5, 8, 13, 21, 34, 55], rtol=1e0, - ) \ No newline at end of file + ) + +def test_example_class(): + """Check the SoftwareGroup class can be instantiated and its class method called""" + sg = SoftwareGroup(people='us', purpose='fun') + sg.long_term_goals()