def test_read_without_spaces(self):
"""
Reading PAZ files where PAZ are not separated by spaces.
Tests uses unrealistic PAZ information.
"""
f = io.StringIO("""CAL1 RJOB LE-3D Z M24 PAZ 010824 0001
2
-4.3982340.48709
-4.39823-4.48709
3
1.2 4.0
-1.09823-3.08709
-1.0982330.08709
0.5""")
p, z, k = paz.read_paz(f)
self.assertAlmostEqual(-4.39823, p[0].real)
self.assertAlmostEqual(40.48709, p[0].imag)
self.assertAlmostEqual(-4.39823, p[1].real)
self.assertAlmostEqual(-4.48709, p[1].imag)
self.assertAlmostEqual(1.2, z[0].real)
self.assertAlmostEqual(4.0, z[0].imag)
self.assertAlmostEqual(-1.09823, z[1].real)
self.assertAlmostEqual(-3.08709, z[1].imag)
self.assertAlmostEqual(-1.09823, z[2].real)
self.assertAlmostEqual(30.08709, z[2].imag)
self.assertAlmostEqual(0.5, k)
f.close()
def _calc_resp(calfile, nfft, sampfreq):
"""
Calculate transfer function of known system.
:type calfile: str
:param calfile: file containing poles, zeros and scale factor for known
system or a dictionary with poles and zeros information (with keys
``'poles'``, ``'zeros'`` and ``'sensitivity'``).
:returns: complex transfer function, array of frequencies
"""
# test if calfile is a paz dict
if isinstance(calfile, dict):
paz = calfile
# or read paz file if a filename is specified
else:
paz = dict()
paz['poles'], paz['zeros'], paz['sensitivity'] = read_paz(calfile)
# calculate transfer function
h, f = paz_to_freq_resp(paz['poles'],
paz['zeros'],
paz['sensitivity'],
1.0 / sampfreq,
nfft,
freq=True)
return h, f
def _calc_resp(calfile, nfft, sampfreq):
"""
Calculate transfer function of known system.
:type calfile: str
:param calfile: file containing poles, zeros and scale factor for known
system
:returns: complex transfer function, array of frequencies
"""
# calculate transfer function
poles, zeros, scale_fac = read_paz(calfile)
h, f = paz_to_freq_resp(poles, zeros, scale_fac, 1.0 / sampfreq, nfft, freq=True)
return h, f
def _calc_resp(calfile, nfft, sampfreq):
"""
Calculate transfer function of known system.
:type calfile: str
:param calfile: file containing poles, zeros and scale factor for known
system
:returns: complex transfer function, array of frequencies
"""
# calculate transfer function
poles, zeros, scale_fac = read_paz(calfile)
h, f = paz_to_freq_resp(poles, zeros, scale_fac, 1.0 / sampfreq,
nfft, freq=True)
return h, f
def _calc_resp(calfile, nfft, sampfreq):
"""
Calculate transfer function of known system.
:type calfile: str
:param calfile: file containing poles, zeros and scale factor for known
system or a dictionary with poles and zeros information (with keys
``'poles'``, ``'zeros'`` and ``'sensitivity'``).
:returns: complex transfer function, array of frequencies
"""
# test if calfile is a paz dict
if isinstance(calfile, dict):
paz = calfile
# or read paz file if a filename is specified
else:
paz = dict()
paz['poles'], paz['zeros'], paz['sensitivity'] = read_paz(calfile)
# calculate transfer function
h, f = paz_to_freq_resp(paz['poles'], paz['zeros'], paz['sensitivity'],
1.0 / sampfreq, nfft, freq=True)
return h, f
def test_read_with_spaces(self):
"""
Reading PAZ files where PAZ are separated by spaces.
"""
f = io.StringIO("""CAL1 RJOB LE-3D Z M24 PAZ 010824 0001
2
-4.39823 4.48709
-4.39823 -4.48709
3
0.0 0.0
0.0 0.0
0.0 0.0
0.4""")
p, z, k = paz.read_paz(f)
self.assertAlmostEqual(-4.39823, p[0].real)
self.assertAlmostEqual(4.48709, p[0].imag)
self.assertAlmostEqual(-4.39823, p[1].real)
self.assertAlmostEqual(-4.48709, p[1].imag)
self.assertEqual([0j, 0j, 0j], z)
self.assertAlmostEqual(0.4, k)
f.close()
