# template file for Lab #6, Task #2
import numpy
import matplotlib.pyplot as p
import lab6
from lab6_1 import freq_res_usr

def zfreqs_to_usr(omega_list):
    pass  # your code here...

if __name__ == '__main__':
    # A quick test for zfreqs_to_usr
    omega_list = [numpy.pi/20.0, numpy.pi/5.0, numpy.pi/2.5]
    usr = zfreqs_to_usr(omega_list)

    # Plot the frequency response of the result
    omega, mag_hejw = freq_res_usr(usr)
    lab6.plot_freq_response(omega, mag_hejw,'three-zero filter')

    p.show()