Mercurial > hg > ltpda
diff testing/utp_1.1/utps/ao/utp_ao_firwhiten.m @ 44:409a22968d5e default
Add unit tests
author | Daniele Nicolodi <nicolodi@science.unitn.it> |
---|---|
date | Tue, 06 Dec 2011 18:42:11 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/testing/utp_1.1/utps/ao/utp_ao_firwhiten.m Tue Dec 06 18:42:11 2011 +0100 @@ -0,0 +1,723 @@ +% UTP_AO_PSD a set of UTPs for the ao/firwhiten method +% +% M Hewitson 06-08-08 +% +% $Id: utp_ao_firwhiten.m,v 1.21 2011/11/16 08:16:08 mauro Exp $ +% + +% <MethodDescription> +% +% The firwhiten method of the ao class computes the spectral density of time-series AOs. +% +% </MethodDescription> + +function results = utp_ao_firwhiten(varargin) + + % Check the inputs + if nargin == 0 + + % Some keywords + class = 'ao'; + mthd = 'firwhiten'; + + results = []; + disp('******************************************************'); + disp(['**** Running UTPs for ' class '/' mthd]); + disp('******************************************************'); + + % Test AOs + [at1,at2,at3,at4,at5,at6,atvec,atmat] = eval(['get_test_objects_' class]); + + % Exception list for the UTPs: + [ple1,ple2,ple3,ple4,ple5,ple6] = get_test_ples(); + + % Run the tests + results = [results utp_01]; % getInfo call + results = [results utp_02]; % Vector input + results = [results utp_03]; % Matrix input + results = [results utp_04]; % List input + results = [results utp_05]; % Test with mixed input + results = [results utp_06]; % Test history is working + results = [results utp_07]; % Test the modify call works + results = [results utp_08]; % Test the data shape + results = [results utp_09]; % Test with complex plist + results = [results utp_10]; % Test the spectral falttening + results = [results utp_11(mthd, at1, ple1)]; % Test plotinfo doesn't disappear + results = [results utp_12(mthd, at1, ple1)]; % Test errors are cleared + + disp('Done.'); + disp('******************************************************'); + + elseif nargin == 1 % Check for UTP functions + if strcmp(varargin{1}, 'isutp') + results = 1; + else + results = 0; + end + else + error('### Incorrect inputs') + end + + %% UTP_01 + + % <TestDescription> + % + % Tests that the getInfo call works for this method. + % + % </TestDescription> + function result = utp_01 + + + % <SyntaxDescription> + % + % Test that the getInfo call works for no sets, all sets, and each set + % individually. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + % Call for no sets + io(1) = eval([class '.getInfo(''' mthd ''', ''None'')']); + % Call for all sets + io(2) = eval([class '.getInfo(''' mthd ''')']); + % Call for each set + for kk=1:numel(io(2).sets) + io(kk+2) = eval([class '.getInfo(''' mthd ''', ''' io(2).sets{kk} ''')']); + end + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check that getInfo call returned an minfo object in all cases. + % 2) Check that all plists have the correct parameters. + % + % </AlgoDescription> + + atest = true; + if stest + % <AlgoCode> + % check we have minfo objects + if isa(io, 'minfo') + prefs = getappdata(0, 'LTPDApreferences'); + defaultWinType = char(prefs.getMiscPrefs.getDefaultWindow); + + %%% SET 'None' + pn = 1; + if ~isempty(io(pn).sets), atest = false; end + if ~isempty(io(pn).plists), atest = false; end + %%% Check all Sets + pn = 2; + if ~any(strcmpi(io(pn).sets, 'Default')), atest = false; end + if numel(io(pn).plists) ~= numel(io(pn).sets), atest = false; end + %%%%%%%%%% SET 'Default' + pn = 3; + if io(pn).plists.nparams ~= 7, atest = false; end + % Check key + if ~io(pn).plists.isparam('nfft'), atest = false; end + if ~io(pn).plists.isparam('bw'), atest = false; end + if ~io(pn).plists.isparam('hc'), atest = false; end + if ~io(pn).plists.isparam('win'), atest = false; end + if ~io(pn).plists.isparam('order'), atest = false; end + if ~io(pn).plists.isparam('firwin'), atest = false; end + if ~io(pn).plists.isparam('ntaps'), atest = false; end + % Check default value + if ~isequal(io(pn).plists.find('nfft'), -1), atest = false; end + if ~isequal(io(pn).plists.find('bw'), 20), atest = false; end + if ~isequal(io(pn).plists.find('hc'), .8), atest = false; end + if ~strcmpi(io(pn).plists.find('win'), defaultWinType), atest = false; end + if ~isequal(io(pn).plists.find('order'), -1), atest = false; end + if ~strcmpi(io(pn).plists.find('win'), defaultWinType), atest = false; end + if ~isequal(io(pn).plists.find('ntaps'), 256), atest = false; end + % Check options + if ~isequal(io(pn).plists.getOptionsForParam('nfft'), {-1}), atest = false; end + if ~isequal(io(pn).plists.getOptionsForParam('bw'), {20}), atest = false; end + if ~isequal(io(pn).plists.getOptionsForParam('hc'), {.8}), atest = false; end + if ~isequal(io(pn).plists.getOptionsForParam('win'), specwin.getTypes), atest = false; end + if ~isequal(io(pn).plists.getOptionsForParam('order'), {-1 0 1 2 3 4 5 6 7 8 9}), atest = false; end + if ~isequal(io(pn).plists.getOptionsForParam('firwin'), specwin.getTypes), atest = false; end + if ~isequal(io(pn).plists.getOptionsForParam('ntaps'), {256}), atest = false; end + end + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_01 + + %% UTP_02 + + % <TestDescription> + % + % Tests that the firwhiten method works with a vector of AOs as input. + % + % </TestDescription> + function result = utp_02 + + % <SyntaxDescription> + % + % Test that the firwhiten method works for a vector of AOs as input. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + avec = [at5 at5 at6]; + N = 512; + fwin = specwin('Hanning', N+1); + [out, outf, outxx] = firwhiten(avec, plist('Ntaps', N, 'FIRwin', fwin)); + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check the number of elements in 'out' + % 2) Check the number of filters (outf) and noise-floor estimates (outxx) + % 3) Check that each output AO contains the correct data. + % + % </AlgoDescription> + + atest = true; + if stest + % <AlgoCode> + % Check we have the correct number of outputs + if numel(out) ~= numel(avec), atest = false; end + if numel(outf) ~= numel(avec), atest = false; end + if numel(outxx) ~= numel(avec), atest = false; end + % Check the output data + prefs = getappdata(0, 'LTPDApreferences'); + swin = char(prefs.getMiscPrefs.getDefaultWindow); + for kk = 1:numel(avec) + pl_psd = plist('Nfft', length(avec(kk).y), 'Win', specwin(swin), 'Order', -1, 'Scale', 'ASD'); + axx = psd(avec(kk), pl_psd); + nxx = smoother(axx, plist('width', 20, 'hc', 0.8)); + w = 1./nxx; + ff = mfir(w, plist('Win', fwin, 'N', N)); + res = filter(avec(kk), ff); + if ~isequal(res.x, out(kk).x), atest = false; end + if ~isequal(res.y, out(kk).y), atest = false; end + end + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_02 + + %% UTP_03 + + % <TestDescription> + % + % Tests that the firwhiten method works with a matrix of AOs as input. + % + % </TestDescription> + function result = utp_03 + + % <SyntaxDescription> + % + % Test that the firwhiten method works for a matrix of AOs as input. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + amat = [at1 at5 at6; at5 at6 at1]; + N = 512; + fwin = specwin('Hanning', N+1); + [out, outf, outxx] = firwhiten(amat, plist('Ntaps', N, 'FIRwin', fwin)); + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check the number of elements in 'out' + % 2) Check the number of filters (outf) and noise-floor estimates (outxx) + % 3) Check that each output AO contains the correct data. + % + % </AlgoDescription> + + atest = true; + if stest + % <AlgoCode> + % Check we have the correct number of outputs + if numel(out) ~= numel(amat), atest = false; end + if numel(outf) ~= numel(amat), atest = false; end + if numel(outxx) ~= numel(amat), atest = false; end + % Check the output data + prefs = getappdata(0, 'LTPDApreferences'); + swin = char(prefs.getMiscPrefs.getDefaultWindow); + for kk = 1:numel(amat) + pl_psd = plist('Nfft', length(amat(kk).y), 'Win', specwin(swin), 'Order', -1, 'Scale', 'ASD'); + axx = psd(amat(kk), pl_psd); + nxx = smoother(axx, plist('width', 20, 'hc', 0.8)); + w = 1./nxx; + ff = mfir(w, plist('Win', fwin, 'N', N)); + res = filter(amat(kk), ff); + if ~isequal(res.x, out(kk).x), atest = false; end + if ~isequal(res.y, out(kk).y), atest = false; end + end + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_03 + + %% UTP_04 + + % <TestDescription> + % + % Tests that the firwhiten method works with a list of AOs as input. + % + % </TestDescription> + function result = utp_04 + + % <SyntaxDescription> + % + % Test that the firwhiten method works for a list of AOs as input. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + N = 512; + fwin = specwin('Hanning', N+1); + [out, outf, outxx] = firwhiten(at1,at5,at6, plist('Ntaps', N, 'FIRwin', fwin)); + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check the number of elements in 'out' + % 2) Check the number of filters (outf) and noise-floor estimates (outxx) + % 3) Check that each output AO contains the correct data. + % + % </AlgoDescription> + + atest = true; + aoin = [at1, at5, at6]; + if stest + % <AlgoCode> + % Check we have the correct number of outputs + if numel(out) ~= 3, atest = false; end + if numel(outf) ~= 3, atest = false; end + if numel(outxx) ~= 3, atest = false; end + % Check the output data + prefs = getappdata(0, 'LTPDApreferences'); + swin = char(prefs.getMiscPrefs.getDefaultWindow); + for kk = 1:numel(aoin) + pl_psd = plist('Nfft', length(aoin(kk).y), 'Win', specwin(swin), 'Order', -1, 'Scale', 'ASD'); + axx = psd(aoin(kk), pl_psd); + nxx = smoother(axx, plist('width', 20, 'hc', 0.8)); + w = 1./nxx; + ff = mfir(w, plist('Win', fwin, 'N', N)); + res = filter(aoin(kk), ff); + if ~isequal(res.x, out(kk).x), atest = false; end + if ~isequal(res.y, out(kk).y), atest = false; end + end + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_04 + + %% UTP_05 + + % <TestDescription> + % + % Tests that the firwhiten method works with a mix of different shaped AOs as + % input. + % + % </TestDescription> + function result = utp_05 + + % <SyntaxDescription> + % + % Test that the firwhiten method works with an input of matrices and vectors + % and single AOs. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + N = 512; + fwin = specwin('Hanning', N+1); + pl = plist('Ntaps', N, 'FIRwin', fwin); + [out, outf, outxx] = firwhiten(at1,[at5 at6],at5,[at5 at1; at6 at1],at6, pl); + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check the number of elements in 'out' + % 2) Check the number of filters (outf) and noise-floor estimates (outxx) + % 3) Check that each output AO contains the correct data. + % + % </AlgoDescription> + + atest = true; + aoin = [at1, reshape([at5 at6], 1, []), at5, reshape([at5 at1; at6 at1], 1, []), at6]; + if stest + % <AlgoCode> + % Check we have the correct number of outputs + if numel(out) ~= 9, atest = false; end + if numel(outf) ~= 9, atest = false; end + if numel(outxx) ~= 9, atest = false; end + % Check the output data + prefs = getappdata(0, 'LTPDApreferences'); + swin = char(prefs.getMiscPrefs.getDefaultWindow); + for kk = 1:numel(aoin) + pl_psd = plist('Nfft', length(aoin(kk).y), 'Win', specwin(swin), 'Order', -1, 'Scale', 'ASD'); + axx = psd(aoin(kk), pl_psd); + nxx = smoother(axx, plist('width', 20, 'hc', 0.8)); + w = 1./nxx; + ff = mfir(w, plist('Win', fwin, 'N', N)); + res = filter(aoin(kk), ff); + if ~isequal(res.x, out(kk).x), atest = false; end + if ~isequal(res.y, out(kk).y), atest = false; end + end + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_05 + + %% UTP_06 + + % <TestDescription> + % + % Tests that the firwhiten method properly applies history. + % + % </TestDescription> + function result = utp_06 + + % <SyntaxDescription> + % + % Test that the result of applying the firwhiten method can be processed back + % to an m-file. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + N = 512; + fwin = specwin('Hanning', N+1); + pl = plist('Ntaps', N, 'FIRwin', fwin); + out = firwhiten(at5, pl); + mout = rebuild(out); + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check that the last entry in the history of 'out' corresponds to + % 'firwhiten'. + % 2) Check that the re-built object is the same object as 'out'. + % + % </AlgoDescription> + + atest = true; + if stest + % <AlgoCode> + % Check the last step in the history of 'out' + if ~strcmp(out.hist.methodInfo.mname, 'firwhiten'), atest = false; end + % Check the re-built object + if ~eq(mout, out, ple4), atest = false; end + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_06 + + %% UTP_07 + + % <TestDescription> + % + % Tests that the firwhiten method can modify the input AO. + % + % </TestDescription> + function result = utp_07 + + % <SyntaxDescription> + % + % Test that the firwhiten method can modify the input AO by calling + % with no output. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + N = 512; + fwin = specwin('Hanning', N+1); + pl = plist('Ntaps', N, 'FIRwin', fwin); + % copy at1 to work with + ain = ao(at1); + % modify ain + aout = ain.firwhiten(pl); + ain.firwhiten(pl); + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check that 'at1' and 'ain' are now different. + % 2) Check that 'ain' is firwhiten(at1). + % + % </AlgoDescription> + + atest = true; + if stest + % <AlgoCode> + % Check that firwhiten modified the input by comparing to the copy + if eq(ao(at1), ain, ple1), atest = false; end + % Check that firwhiten doesn't modified the input for the function notation + if ~eq(aout, ain, ple1), atest = false; end + % Check that the modified input is the firwhiten of the copy + % Check the output data of ain + prefs = getappdata(0, 'LTPDApreferences'); + swin = char(prefs.getMiscPrefs.getDefaultWindow); + pl_psd = plist('Nfft', length(at1.y), 'Win', specwin(swin), 'Order', -1, 'Scale', 'ASD'); + axx = psd(at1, pl_psd); + nxx = smoother(axx, plist('width', 20, 'hc', 0.8)); + w = 1./nxx; + ff = mfir(w, plist('Win', fwin, 'N', N)); + res = filter(at1, ff); + if ~isequal(res.x, ain.x), atest = false; end + if ~isequal(res.y, ain.y), atest = false; end + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_07 + + %% UTP_08 + + % <TestDescription> + % + % Tests that the firwhiten method keeps the data shape of the input object. + % + % </TestDescription> + function result = utp_08 + + % <SyntaxDescription> + % + % Test that the firwhiten method keeps the data shape of the input object. The + % input AO must be an AO with row data and an AO with column data. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + N = 512; + fwin = specwin('Hanning', N+1); + pl = plist('Ntaps', N, 'FIRwin', fwin); + out1 = firwhiten(at5, pl); + out2 = firwhiten(at6, pl); + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check that the shpe of the data doesn't change. + % + % </AlgoDescription> + + atest = true; + if stest + % <AlgoCode> + % Check the shape of the output data + if size(out1.data.y,2) ~= 1, atest = false; end + if size(out2.data.y,1) ~= 1, atest = false; end + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_08 + + %% UTP_09 + + % <TestDescription> + % + % Tests that the firwhiten method with a complex plist. + % + % </TestDescription> + function result = utp_09 + + % <SyntaxDescription> + % + % Test that the result of applying the firwhiten method with a complex plist + % can be processed back to a m-file. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + N = 512; + Nfft = 100; + order = 1; + fwin = specwin('Hanning', N+1); + swin = specwin('Kaiser', 10, 100); + pl = plist('Ntaps', N, 'FIRwin', fwin, 'win', swin, 'Nfft', Nfft, 'order', order); + [out, outf, outxx] = firwhiten(at5, pl); + mout = rebuild(out); + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Check the output data + % 2) Check the output filter + % 3) Check the noise-floor estimates + % 4) Check that the re-built object is the same object as 'out'. + % + % </AlgoDescription> + + atest = true; + if stest + % <AlgoCode> + % Check the last step in the history of 'out' + if ~strcmp(out.hist.methodInfo.mname, 'firwhiten'), atest = false; end + % Check the re-built object + if ~eq(mout, out, ple2), atest = false; end + + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_09 + + %% UTP_10 + + % <TestDescription> + % + % Test the spectral falttening capability of firwhiten method. + % + % </TestDescription> + function result = utp_10 + + % <SyntaxDescription> + % + % Test that the application of the firwhiten method enhances the + % spectral flatness of input data. + % + % </SyntaxDescription> + + try + % <SyntaxCode> + % Making test data + fs = at5.fs; + pzm = pzmodel(2, [pz(0.1, 2) pz(0.5, 1)] , [pz(1) pz(2.55, 2)]); + ft = miir(pzm,plist('fs',fs)); + af = filter(at5, ft); % Colored noise + + % Whitening + N = 512; + Nfft = 100; + order = 1; + fwin = specwin('Hanning', N+1); + swin = specwin('Kaiser', 10, 100); + pl = plist('Ntaps', N, 'FIRwin', fwin, 'win', swin, 'Nfft', Nfft, 'order', order); + out = firwhiten(af, pl); + + % </SyntaxCode> + stest = true; + catch err + disp(err.message) + stest = false; + end + + % <AlgoDescription> + % + % 1) Calculate PSD of input and whitened data + % 2) Compare relative spectral flatness coefficients + % + % </AlgoDescription> + + atest = true; + if stest + % <AlgoCode> + % Check the output data +% pl_psd = plist('Nfft', Nfft, 'Win', swin, 'Order', order, 'Scale', 'PSD'); + axx = af.psd; + awxx = out.psd; + + % Claculating flatness + sf1 = utils.math.spflat(axx.data.y); + sf2 = utils.math.spflat(awxx.data.y); + + % Checking flatness + if sf1>sf2, atest = false; end + + % </AlgoCode> + else + atest = false; + end + + % Return a result structure + result = utp_prepare_result(atest, stest, dbstack, mfilename); + end % END UTP_10 + + + +end