#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# John A. Marohn (jam99@cornell.edu)
# 2014/06/28
"""
Tests for the demodulate module.
"""
# if a function is *expected *to fail, then
#
# @unittest.expectedFailure
# def test_that_fails():
#
from freqdemod.demodulate import Signal
import unittest
import numpy as np
# import h5py
[docs]class InitLoadSaveTests(unittest.TestCase):
"""
Make sure the *Signal* object is set up correctly.
"""
[docs] def setUp(self):
"""
Create an trial *Signal* object
"""
self.s = Signal('.InitLoadSaveTests_1.h5')
self.s.load_nparray(np.arange(3),"x","nm",10E-6)
[docs] def test_report(self):
"""Initialize Signal object"""
self.assertEqual(self.s.report[0],'HDF5 file .InitLoadSaveTests_1.h5 created in core memory')
[docs] def test_x(self):
"""Check x-array data."""
self.assertTrue(np.allclose(self.s.f['x'],10E-6*np.array([0, 1, 2]), rtol=1e-05, atol=1e-08))
[docs] def test_y(self):
"""Check y-array data."""
self.assertTrue(np.allclose(self.s.f['y'],np.array([0, 1, 2]), rtol=1e-05, atol=1e-08))
[docs] def test_close(self):
"""Verify closed object by testing one of the attributes"""
self.s.close()
self.snew = Signal()
self.snew.open('.InitLoadSaveTests_1.h5')
# print out the contents of the file nicely
report = []
for key, val in self.snew.f.attrs.items():
report.append("{0}: {1}".format(key, val))
for item in self.snew.f:
report.append("{}".format(self.snew.f[item].name))
for key, val in self.snew.f[item].attrs.items():
report.append(" {0}: {1}".format(key, val))
report_string = "\n".join(report)
print "\nObjects in file .InitLoadSaveTests_1.h5"
print report_string
# test one of the attributes
self.assertTrue(self.snew.f.attrs['source'],'demodulate.py')
[docs] def tearDown(self):
"""Close the h5 files before the next iteration."""
try:
self.s.f.close()
except:
pass
try:
self.snew.f.close()
except:
pass
[docs]class MaskTests(unittest.TestCase):
[docs] def setUp(self):
"""
Create a trial *Signal* object
"""
self.s = Signal('.InitLoadSaveTests_1.h5')
self.s.load_nparray(np.arange(60000),"x","nm",10E-6)
[docs] def test_binarate_1(self):
"""Binarate mask middle; test length is 2^n"""
self.s.time_mask_binarate("middle")
m = self.s.f['workup/time/mask/binarate']
self.assertEqual(np.count_nonzero(m),32*1024)
[docs] def test_binarate_2(self):
"""Binarate mask start; test length is 2^n"""
self.s.time_mask_binarate("start")
m = self.s.f['workup/time/mask/binarate']
self.assertEqual(np.count_nonzero(m),32*1024)
[docs] def test_binarate_3(self):
"""Binarate mask end test length is 2^n"""
self.s.time_mask_binarate("end")
m = self.s.f['workup/time/mask/binarate']
self.assertEqual(np.count_nonzero(m),32*1024)
[docs] def test_binarate_4(self):
"""If we have not called binarate, then workup/time/mask/binarate does not exist"""
self.assertEqual(self.s.f.__contains__('workup/time/mask/binarate'),False)
[docs] def test_binarate_5(self):
"""If we have called binarate, then workup/time/mask/binarate does exist"""
self.s.time_mask_binarate("middle")
self.assertEqual(self.s.f.__contains__('workup/time/mask/binarate'),True)
[docs] def tearDown(self):
"""Close the h5 files before the next iteration."""
try:
self.s.f.close()
except:
pass
[docs]class FFTTests(unittest.TestCase):
[docs] def setUp(self):
"""
Create an trial *Signal* object
"""
fd = 50.0E3 # digitization frequency
f0 = 5.00E3 # signal frequency
nt = 512 # number of signal points
dt = 1/fd
t = dt*np.arange(nt)
s = 1.0*np.sin(2*np.pi*f0*t)
self.s = Signal('.FFTTests_1.h5')
self.s.load_nparray(s,"x","nm",dt)
self.s.time_mask_binarate("middle")
self.s.time_window_cyclicize(10*dt)
self.s.fft()
self.s.freq_filter_Hilbert_complex()
[docs] def testfft_1(self):
"""FFT: test that the resulting data is complex"""
first_point = self.s.f['workup/freq/FT'][0]
self.assertEqual(isinstance(first_point, complex),True)
[docs] def testfft_2(self):
"""FFT: test that the complex Hilbert transform filter is real"""
first_point = self.s.f['workup/freq/filter/Hc'][0]
self.assertEqual(isinstance(first_point, complex),False)
[docs] def testfft_3(self):
"""FFT: test the complex Hilbert transform filter near freq = 0"""
freq = self.s.f['workup/freq/freq'][:]
index = np.roll(freq == 0,-1) + np.roll(freq == 0,0) + np.roll(freq == 0,1)
filt = self.s.f['workup/freq/filter/Hc'][index]
self.assertTrue(np.allclose(filt,np.array([0, 1, 2])))
[docs] def tearDown(self):
"""Close the h5 files before the next iteration."""
try:
self.s.f.close()
except:
pass
[docs]class MiscTests(unittest.TestCase):
[docs] def test_array_middle_1(self):
"""Misc: mean time (n odd)"""
n = 5
t = np.arange(n) # [0,1,2,3,4] => mean 2 -> t[2]
self.assertEqual(np.mean(t,axis=0),2.0)
[docs] def test_array_middle_2(self):
"""Misc: mean time (n even)"""
n = 6
t = np.arange(n) # [0,1,2,3,4,5] => mean 2.5 => (t[2]+t[3])/2.0
self.assertEqual(np.mean(t,axis=0),2.5)