def amplitude_response(spec_real, spec_imag, db=True):
size = len(spec_real)
amp = [0] * size
for i in range(0, size):
amp[i] = dcx.modulus(spec_real[i], spec_imag[i]) / (size/2)
if db:
amp[i] = dcv.lin2db(amp[i])
return amp
def test_one_zero(self):
self.assertEqual(dc.modulus(1.0, 0.0), 1.0)
def test_sqrt_3_on_2(self):
self.assertAlmostEqual(dc.modulus(math.sqrt(3) / 2.0, 1.0 / 2.0), 1.0)
def test_zero_zero(self):
self.assertEqual(dc.modulus(0.0, 0.0), 0.0)
def test_one_one(self):
self.assertEqual(dc.modulus(1.0, 1.0), math.sqrt(2))
def test_sqrt_2_on_2(self):
self.assertEqual(dc.modulus(math.sqrt(2) / 2.0,
math.sqrt(2) / 2.0), 1.0)
def test_zero_zero(self):
self.assertEqual(dc.modulus(0.0, 0.0), 0.0)
def test_zero_one(self):
self.assertEqual(dc.modulus(0.0, 1.0), 1.0)
def test_sqrt_3_on_2(self):
self.assertAlmostEqual(dc.modulus(math.sqrt(3) / 2.0, 1.0 / 2.0), 1.0)
def test_sqrt_2_on_2(self):
self.assertEqual(dc.modulus(math.sqrt(2) / 2.0, math.sqrt(2) / 2.0), 1.0)
def test_one_one(self):
self.assertEqual(dc.modulus(1.0, 1.0), math.sqrt(2))
def test_zero_one(self):
self.assertEqual(dc.modulus(0.0, 1.0), 1.0)
def test_one_zero(self):
self.assertEqual(dc.modulus(1.0, 0.0), 1.0)
