1
0
Fork 0
This repository has been archived on 2021-01-12. You can view files and clone it, but cannot push or open issues or pull requests.
uni-m.cds-num-met/week1/test_ex2.py

45 lines
1.3 KiB
Python
Raw Normal View History

2020-02-12 12:26:58 +01:00
#!/usr/bin/env python3
# Test Lagrange Polynomial Interpolation Implementation
#------------------------------------------------------
from ex2 import my_lagrange, my_lagrange_polynomials
import numpy as np
def test_import():
"""
Test the import
"""
assert my_lagrange is not None
assert my_lagrange_polynomials is not None
def test_my_lagrange_polynomials():
"""
Test the my_lagrange_polynomials function
"""
# Take Numpy Array
assert (my_lagrange_polynomials(np.array([0,1]), 1) == np.array([0, 1])).all()
# Take Python List
assert (my_lagrange_polynomials([0,1], 1) == np.array([0, 1])).all()
# Changing the order of x_values changes the order of the returned values
assert (my_lagrange_polynomials([2,4,6,8], 5) == np.array([-0.0625, 0.5625, 0.5625, -0.0625])).all()
assert (my_lagrange_polynomials([6,8,2,4], 5) == np.array([0.5625, -0.0625, -0.0625, 0.5625])).all()
def test_my_lagrange():
"""
Test the my_lagrange function
"""
# Take x_i as a single number
assert (my_lagrange([1, 2], [1, 2], 1) == np.array([1])).all()
assert (my_lagrange([1, 2], [1, 2], [1, 1.5, 2]) == np.array([1, 1.5, 2])).all()
if __name__ == "__main__":
import pytest
print("pyTesting ", __file__)
pytest.main([__file__])