Merge pull request #124 from jaidevd/jd-test-postproc

Add a couple of tests for the postprocessing functions

continuous_integration/requirements.txt

@@ -3,3 +3,4 @@ scipy
 matplotlib
 nose
 libgfortran
+scikit-image

tftb/processing/postprocessing.py

@@ -23,8 +23,8 @@ def hough_transform(image, m=None, n=None):
     :type image:
     :type m:
     :type n:
-:return:
-:rtype:
+    :return:
+    :rtype:
     """
     xmax, ymax = image.shape
     if m is None:
@@ -78,8 +78,8 @@ def renyi_information(tfr, timestamps=None, freq=None, alpha=3.0):
     :type timestamps:
     :type freq:
     :type alpha:
-:return:
-:rtype:
+    :return:
+    :rtype:
     """
     if alpha == 1 and tfr.min().min() < 0:
         raise ValueError("Distribution with negative values not allowed.")
@@ -109,8 +109,8 @@ def ideal_tfr(iflaws, timestamps=None, n_fbins=None):
     :type iflaws:
     :type timestamps:
     :type n_fbins:
-:return:
-:rtype:
+    :return:
+    :rtype:
     """
     ifrow, ifcol = iflaws.shape
     if timestamps is None:
@@ -141,8 +141,8 @@ def friedman_density(tfr, re_mat, timestamps=None):
     :type tfr:
     :type re_mat:
     :type timestamps:
-:return:
-:rtype:
+    :return:
+    :rtype:
     """
     tfrrow, tfrcol = tfr.shape
     hatrow, hatcol = re_mat.shape
@@ -174,8 +174,8 @@ def ridges(tfr, re_mat, timestamps=None, method='rsp'):
     :type re_mat:
     :type timestamps:
     :type method:
-:return:
-:rtype:
+    :return:
+    :rtype:
     """
     method = method.lower()
     tfrrow, tfrcol = tfr.shape

tftb/processing/tests/test_postprocessing.py

@@ -0,0 +1,54 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+# vim:fenc=utf-8
+#
+# Cube26 product code
+#
+# (C) Copyright 2015 Cube26 Software Pvt Ltd
+# All right reserved.
+#
+# This file is confidential and NOT open source.  Do not distribute.
+#
+
+"""
+Tests for tftb.processing.postprocessing
+"""
+
+from tftb.tests.base import TestBase
+from tftb.generators import atoms, fmlin
+from tftb.processing import WignerVilleDistribution
+from tftb.processing import postprocessing as pproc
+from skimage.transform import hough_line_peaks, hough_line
+import numpy as np
+import unittest
+
+
+class TestPostprocessing(TestBase):
+
+    def test_renyi_information(self):
+        """Check if Renyi entropy computation is correct."""
+        sig = atoms(128, np.array([[64., 0.25, 20., 1.]]))
+        tfr, _, _ = WignerVilleDistribution(sig).run()
+        R1 = pproc.renyi_information(tfr)
+        sig = atoms(128, np.array([[32., 0.25, 20., 1.],
+                                   [96., 0.25, 20., 1.]]))
+        tfr, _, _ = WignerVilleDistribution(sig).run()
+        R2 = pproc.renyi_information(tfr)
+        self.assertAlmostEqual(R2 - R1, 0.98, places=1)
+
+    def test_ideal_tfr(self):
+        """Test if the ideal TFR can be found using the instantaneous frequency
+        laws."""
+        _, iflaw1 = fmlin(128, 0.0, 0.2)
+        _, iflaw2 = fmlin(128, 0.3, 0.5)
+        iflaws = np.c_[iflaw1, iflaw2].T
+        tfr, _, _ = pproc.ideal_tfr(iflaws)
+        tfr[tfr == 1] = 255
+        tfr = tfr.astype(np.uint8)
+        hspace, angles, dists = hough_line(tfr)
+        for x in hough_line_peaks(hspace, angles, dists):
+            self.assertEqual(len(x), 2)
+
+
+if __name__ == '__main__':
+    unittest.main()