<s> import unittest <EOL> import numpy as np <EOL> import deltasigma as ds <EOL> from deltasigma . _utils import cplxpair <EOL> class TestCalculateTF ( unittest . TestCase ) : <EOL> """<STR_LIT>""" <EOL> def setUp ( self ) : <EOL> ABCD = [ [ <NUM_LIT:1.0> , <NUM_LIT:0.0> , <NUM_LIT:0.0> , <NUM_LIT> , - <NUM_LIT> ] , <EOL> [ <NUM_LIT> , <NUM_LIT> , - <NUM_LIT> , <EOL> <NUM_LIT:0.0> , <NUM_LIT> ] , <EOL> [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT:0.0> , - <NUM_LIT> ] , <EOL> [ <NUM_LIT:0.0> , <NUM_LIT:0.0> , <NUM_LIT:1.0> , <NUM_LIT:0.0> , <NUM_LIT:0.0> ] ] <EOL> ABCD = np . array ( ABCD ) <EOL> ntf , stf = ds . calculateTF ( ABCD ) <EOL> ntf_zeros , ntf_poles , _ = ntf <EOL> stf_zeros , stf_poles , _ = stf <EOL> mntf_poles = np . array ( ( <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> ) ) <EOL> mntf_zeros = np . array ( ( <NUM_LIT> + <NUM_LIT> , <EOL> <NUM_LIT> - <NUM_LIT> , <EOL> <NUM_LIT> + <NUM_LIT> ) ) <EOL> mstf_zeros = np . array ( ( - <NUM_LIT> , ) ) <EOL> mstf_poles = np . array ( ( <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> ) ) <EOL> self . ntf_zeros , self . mntf_zeros = ( cplxpair ( ntf_zeros ) , <EOL> cplxpair ( mntf_zeros ) ) <EOL> self . stf_zeros , self . mstf_zeros = ( cplxpair ( stf_zeros ) , <EOL> cplxpair ( mstf_zeros ) ) <EOL> self . ntf_poles , self . mntf_poles = ( cplxpair ( ntf_poles ) , <EOL> cplxpair ( mntf_poles ) ) <EOL> self . stf_poles , self . mstf_poles = ( cplxpair ( stf_poles ) , <EOL> cplxpair ( mstf_poles ) ) <EOL> def test_calculateTF1 ( self ) : <EOL> """<STR_LIT>""" <EOL> self . assertTrue ( np . allclose ( self . ntf_zeros , self . mntf_zeros , rtol = <NUM_LIT> , <EOL> atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( self . ntf_poles , self . mntf_poles , rtol = <NUM_LIT> , <EOL> atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( self . stf_zeros , self . mstf_zeros , rtol = <NUM_LIT> , <EOL> atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( self . stf_poles , self . mstf_poles , rtol = <NUM_LIT> , <EOL> atol = <NUM_LIT> ) ) <EOL> def test_calculateTF2 ( self ) : <EOL> """<STR_LIT>""" <EOL> ABCD = np . array ( [ [ <NUM_LIT:1.> , <NUM_LIT:1.> , - <NUM_LIT:1.> ] , <EOL> [ <NUM_LIT:1.> , <NUM_LIT:0.> , <NUM_LIT:0.> ] ] ) <EOL> k = <NUM_LIT:1.> <EOL> ntf , stf = ds . calculateTF ( ABCD , k ) <EOL> ntf_zeros , ntf_poles , ntf_gain = ntf <EOL> stf_zeros , stf_poles , stf_gain = stf <EOL> self . assertTrue ( np . allclose ( stf_poles , [ <NUM_LIT:0.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( not len ( stf_zeros ) ) <EOL> self . assertTrue ( np . allclose ( stf_gain , <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntf_poles , [ <NUM_LIT:0.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntf_zeros , [ <NUM_LIT:1.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntf_gain , <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> def test_calculateTF3 ( self ) : <EOL> """<STR_LIT>""" <EOL> ABCD = np . array ( [ [ <NUM_LIT:1.> , <NUM_LIT:1.> , - <NUM_LIT:1.> ] , <EOL> [ <NUM_LIT:1.> , <NUM_LIT:0.> , <NUM_LIT:0.> ] ] ) <EOL> ntf , stf = ds . calculateTF ( ABCD ) <EOL> ntf_zeros , ntf_poles , ntf_gain = ntf <EOL> stf_zeros , stf_poles , stf_gain = stf <EOL> self . assertTrue ( np . allclose ( stf_poles , [ <NUM_LIT:0.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( not len ( stf_zeros ) ) <EOL> self . assertTrue ( np . allclose ( stf_gain , <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntf_poles , [ <NUM_LIT:0.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntf_zeros , [ <NUM_LIT:1.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntf_gain , <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> def test_calculateTF4 ( self ) : <EOL> """<STR_LIT>""" <EOL> ABCD = [ [ <NUM_LIT:1> , <NUM_LIT:1> , - <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> [ <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> ] , <EOL> [ <NUM_LIT:1> , <NUM_LIT:0> , - <NUM_LIT:1> , <NUM_LIT:0> ] ] <EOL> ABCD = np . array ( ABCD , dtype = np . float_ ) <EOL> k = [ <NUM_LIT:1.> , <NUM_LIT:1.> ] <EOL> ntfs , stfs = ds . calculateTF ( ABCD , k = k ) <EOL> self . assertTrue ( np . allclose ( stfs [ <NUM_LIT:0> ] [ <NUM_LIT:1> ] , [ <NUM_LIT:0.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( not len ( stfs [ <NUM_LIT:0> ] [ <NUM_LIT:0> ] ) ) <EOL> self . assertTrue ( np . allclose ( stfs [ <NUM_LIT:0> ] [ <NUM_LIT:2> ] , <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( stfs [ <NUM_LIT:1> ] [ <NUM_LIT:1> ] , [ <NUM_LIT:0.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( not len ( stfs [ <NUM_LIT:1> ] [ <NUM_LIT:0> ] ) ) <EOL> self . assertTrue ( np . allclose ( stfs [ <NUM_LIT:1> ] [ <NUM_LIT:2> ] , <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntfs [ <NUM_LIT:0> , <NUM_LIT:0> ] [ <NUM_LIT:1> ] , [ <NUM_LIT:0.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntfs [ <NUM_LIT:0> , <NUM_LIT:0> ] [ <NUM_LIT:0> ] , [ <NUM_LIT:1.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntfs [ <NUM_LIT:0> , <NUM_LIT:0> ] [ <NUM_LIT:2> ] , <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntfs [ <NUM_LIT:1> , <NUM_LIT:0> ] [ <NUM_LIT:1> ] , [ <NUM_LIT:0.> ] , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( not len ( ntfs [ <NUM_LIT:1> , <NUM_LIT:0> ] [ <NUM_LIT:0> ] ) ) <EOL> self . assertTrue ( np . allclose ( ntfs [ <NUM_LIT:1> , <NUM_LIT:0> ] [ <NUM_LIT:2> ] , - <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( not len ( ntfs [ <NUM_LIT:1> , <NUM_LIT:1> ] [ <NUM_LIT:0> ] ) ) <EOL> self . assertTrue ( not len ( ntfs [ <NUM_LIT:1> , <NUM_LIT:1> ] [ <NUM_LIT:1> ] ) ) <EOL> self . assertTrue ( np . allclose ( ntfs [ <NUM_LIT:1> , <NUM_LIT:1> ] [ <NUM_LIT:2> ] , <NUM_LIT:1.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) <EOL> self . assertTrue ( np . allclose ( ntfs [ <NUM_LIT:0> , <NUM_LIT:1> ] [ <NUM_LIT:2> ] , <NUM_LIT:0.> , rtol = <NUM_LIT> , atol = <NUM_LIT> ) ) </s>
