comments/variables-cleanup

[openpower-isa.git] / src / openpower / decoder / isa / fastdct-test.py

diff --git a/src/openpower/decoder/isa/fastdct-test.py b/src/openpower/decoder/isa/fastdct-test.py
index b1db7803b8728c9fae52bba3e6f2daa1fee3f8f3..199ea3993837579fe9d7427df84e7464aa0050f7 100644 (file)
--- a/src/openpower/decoder/isa/fastdct-test.py
+++ b/src/openpower/decoder/isa/fastdct-test.py
@@ -28,7 +28,7 @@ import fastdctlee, naivedct, remap_dct_yield
 class FastDctTest(unittest.TestCase):
 
     def tst_fast_dct_lee_vs_naive(self):
-        for i in range(3, 10):
+        for i in range(3, 4):
             n = 2**i
             vector = FastDctTest.nonrandom_vector(n)
             expect = naivedct.transform(vector)
@@ -36,21 +36,24 @@ class FastDctTest(unittest.TestCase):
             actual = fastdctlee.transform2(vector)
             self.assertListAlmostEqual(actual, expect)
             expect = naivedct.inverse_transform(vector)
-            actual = fastdctlee.inverse_transform(vector)
+            original = fastdctlee.inverse_transform2(vector)
+            actual = fastdctlee.inverse_transform_iter(vector)
             self.assertListAlmostEqual(actual, expect)
 
     def test_yield_dct_lee_vs_naive(self):
-        for i in range(3, 10):
+        for i in range(2, 10):
             n = 2**i
             vector = FastDctTest.nonrandom_vector(n)
             expect = fastdctlee.transform2(vector)
             actual = remap_dct_yield.transform2(vector)
             self.assertListAlmostEqual(actual, expect)
             expect = naivedct.inverse_transform(vector)
-            actual = fastdctlee.inverse_transform(vector)
+            actual = fastdctlee.inverse_transform_iter(vector)
+            self.assertListAlmostEqual(actual, expect)
+            actual = remap_dct_yield.inverse_transform2(vector)
             self.assertListAlmostEqual(actual, expect)
 
-    def notest_fast_dct_lee_invertibility(self):
+    def tst_fast_dct_lee_invertibility(self):
         for i in range(1, 10):
             n = 2**i
             vector = FastDctTest.random_vector(n)
@@ -59,6 +62,15 @@ class FastDctTest(unittest.TestCase):
             temp = [(val * 2.0 / n) for val in temp]
             self.assertListAlmostEqual(vector, temp)
 
+    def tst_yield_fast_dct_lee_invertibility(self):
+        for i in range(1, 10):
+            n = 2**i
+            vector = FastDctTest.random_vector(n)
+            temp = remap_dct_yield.transform2(vector)
+            temp = fastdctlee.inverse_transform(temp)
+            temp = [(val * 2.0 / n) for val in temp]
+            self.assertListAlmostEqual(vector, temp)
+
     def assertListAlmostEqual(self, actual, expect):
         self.assertEqual(len(actual), len(expect))
         for (x, y) in zip(actual, expect):