Remove codec tests for codes with N = 2048

Author: grigory

python_polar_coding/polar_codes/__init__.py

@@ -0,0 +1,5 @@
+from .fast_ssc import FastSSCPolarCodec
+from .g_fast_ssc import GFastSSCPolarCodec
+from .rc_scan import RCSCANPolarCodec
+from .sc import SCPolarCodec
+from .sc_list import SCListPolarCodec

python_polar_coding/polar_codes/g_fast_ssc/__init__.py

@@ -1,3 +1,3 @@
-from .codec import GeneralizedFastSSCPolarCodec
-from .decoder import GeneralizedFastSSCDecoder
+from .codec import GFastSSCPolarCodec
+from .decoder import GFastSSCDecoder
 from .node import GFastSSCNode

python_polar_coding/polar_codes/g_fast_ssc/codec.py

@@ -2,16 +2,16 @@
 
 from python_polar_coding.polar_codes.fast_ssc import FastSSCPolarCodec
 
-from .decoder import GeneralizedFastSSCDecoder
+from .decoder import GFastSSCDecoder
 
 
-class GeneralizedFastSSCPolarCodec(FastSSCPolarCodec):
+class GFastSSCPolarCodec(FastSSCPolarCodec):
     """Generalized Fast SSC code.
 
     Based on: https://arxiv.org/pdf/1804.09508.pdf
 
     """
-    decoder_class = GeneralizedFastSSCDecoder
+    decoder_class = GFastSSCDecoder
 
     def __init__(
             self,

python_polar_coding/polar_codes/g_fast_ssc/decoder.py

@@ -5,10 +5,10 @@
 from .node import GFastSSCNode
 
 
-class GeneralizedFastSSCDecoder(FastSSCDecoder):
+class GFastSSCDecoder(FastSSCDecoder):
     node_class = GFastSSCNode
 
-    def __init__(self, n: int, mask: np.array, AF: int = 1):
+    def __init__(self, n: int, mask: np.array, AF: int = 0):
         self.AF = AF
         super().__init__(n=n, mask=mask)
 

python_polar_coding/simulation/simulation.py

@@ -5,12 +5,12 @@
 from ..channels import SimpleAWGNChannel
 from ..modems import SimpleBPSKModem
 from ..polar_codes import (
-    FastSCANCodec,
     FastSSCPolarCodec,
-    GeneralizedFastSSCPolarCodec,
-    GFastSCANCodec,
+    GFastSSCPolarCodec,
     RCSCANPolarCodec,
 )
+from ..polar_codes.fast_scan import FastSCANCodec
+from ..polar_codes.g_fast_scan import GFastSCANCodec
 from . import functions, http
 
 
@@ -28,6 +28,15 @@ class CodeTypes:
         FAST_SCAN,
         G_FAST_SCAN,
     ]
+    SCAN = [
+        RC_SCAN,
+        FAST_SCAN,
+        G_FAST_SCAN,
+    ]
+    GENERALIZED = [
+        G_FAST_SSC,
+        G_FAST_SCAN,
+    ]
 
 
 class ChannelTypes:
@@ -37,7 +46,7 @@ class ChannelTypes:
 CODE_MAP = {
     CodeTypes.FAST_SSC: FastSSCPolarCodec,
     CodeTypes.RC_SCAN: RCSCANPolarCodec,
-    CodeTypes.G_FAST_SSC: GeneralizedFastSSCPolarCodec,
+    CodeTypes.G_FAST_SSC: GFastSSCPolarCodec,
     CodeTypes.FAST_SCAN: FastSCANCodec,
     CodeTypes.G_FAST_SCAN: GFastSCANCodec,
 }
@@ -101,10 +110,14 @@ def simulate_from_params(url: str):
         code_params=experiment,
     )
 
-    result_log = (f'Result: {result}\n'
-                  f'{code_type.upper()} ({experiment["N"]},{experiment["K"]})')
-    if code_type == CodeTypes.RC_SCAN:
+    result_log = (
+        f'Result: {result}\n'
+        f'{code_type.upper()} ({experiment["N"]},{experiment["K"]})'
+    )
+    if code_type in CodeTypes.SCAN:
         result_log += f', I = {experiment["I"]}'
+    if code_type in CodeTypes.GENERALIZED:
+        result_log += f', AF = {experiment["AF"]}'
     print(result_log)
 
     resp = http.save_result(
@@ -115,7 +128,7 @@ def simulate_from_params(url: str):
         channel_type=channel_type,
         cls=cls,
     )
-    print(f'Status: {resp.status_code}: {resp.json()}')
+    print(f'Status {resp.status_code}: {resp.json()}')
 
 
 def simulate_multi_core(experiments: int, url: str):
@@ -124,8 +137,10 @@ def simulate_multi_core(experiments: int, url: str):
     print(f'Workers: {workers}; Number of experiments: {experiments}')
 
     with futures.ProcessPoolExecutor(max_workers=workers) as ex:
-        run_tasks = {ex.submit(simulate_from_params, *(url, )): (url, )
-                     for _ in range(experiments)}
+        run_tasks = {
+            ex.submit(simulate_from_params, *(url, )): (url, )
+            for _ in range(experiments)
+        }
         for future in futures.as_completed(run_tasks):
             try:
                 future.result()

python_polar_coding/tests/test_fast_scan/test_codec.py

@@ -31,33 +31,6 @@ class TestFastSCANCodec_1024_768(BasicVerifyPolarCode, TestCase):
     }
 
 
-class TestFastSCANCodec_2048_512(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 512,
-        'I': 1,
-    }
-
-
-class TestFastSCANCodec_2048_1024(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 1024,
-        'I': 1,
-    }
-
-
-class TestFastSCANCodec_2048_1536(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 1536,
-        'I': 1,
-    }
-
-
 # Iterations 2
 
 
@@ -88,33 +61,6 @@ class TestFastSCANCodec_1024_768_iter_2(BasicVerifyPolarCode, TestCase):
     }
 
 
-class TestFastSCANCodec_2048_512_iter_2(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 512,
-        'I': 2,
-    }
-
-
-class TestFastSCANCodec_2048_1024_iter_2(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 1024,
-        'I': 2,
-    }
-
-
-class TestFastSCANCodec_2048_1536_iter_2(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 1536,
-        'I': 2,
-    }
-
-
 # Iterations 4
 
 
@@ -143,30 +89,3 @@ class TestFastSCANCodec_1024_768_iter_4(BasicVerifyPolarCode, TestCase):
         'K': 768,
         'I': 4,
     }
-
-
-class TestFastSCANCodec_2048_512_iter_4(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 512,
-        'I': 4,
-    }
-
-
-class TestFastSCANCodec_2048_1024_iter_4(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 1024,
-        'I': 4,
-    }
-
-
-class TestFastSCANCodec_2048_1536_iter_4(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSCANCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 1536,
-        'I': 4,
-    }

python_polar_coding/tests/test_fast_ssc/test_codec.py

@@ -26,27 +26,3 @@ class TestFastSSCCode_1024_768(BasicVerifyPolarCode, TestCase):
         'N': 1024,
         'K': 768,
     }
-
-
-class TestFastSSCCode_2048_512(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSSCPolarCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 512,
-    }
-
-
-class TestFastSSCCode_2048_1024(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSSCPolarCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 1024,
-    }
-
-
-class TestFastSSCCode_2048_1536(BasicVerifyPolarCode, TestCase):
-    polar_code_class = FastSSCPolarCodec
-    code_parameters = {
-        'N': 2048,
-        'K': 1536,
-    }