Python 2 compatibility.

Author: BastienTr

commpy/channels.py

@@ -18,14 +18,16 @@
 
 """
 
+from __future__ import division, print_function  # Python 2 compatibility
+
 from numpy import complex, abs, sqrt, sum, zeros, identity, hstack, einsum, trace, kron, absolute
 from numpy.random import randn, random, standard_normal
 from scipy.linalg import sqrtm
 
 __all__ = ['SISOFlatChannel', 'MIMOFlatChannel', 'bec', 'bsc', 'awgn']
 
 
-class _FlatChannel:
+class _FlatChannel(object):
 
     def __init__(self):
         self.noises = None
@@ -172,7 +174,7 @@ def nb_rx(self):
         return 1
 
     def __init__(self, noise_std=None, fading_param=(1, 0)):
-        super().__init__()
+        super(SISOFlatChannel, self).__init__()
         self.noise_std = noise_std
         self.fading_param = fading_param
 
@@ -317,7 +319,7 @@ class MIMOFlatChannel(_FlatChannel):
     """
 
     def __init__(self, nb_tx, nb_rx, noise_std=None, fading_param=None):
-        super().__init__()
+        super(MIMOFlatChannel, self).__init__()
         self.nb_tx = nb_tx
         self.nb_rx = nb_rx
         self.noise_std = noise_std

commpy/tests/test_channels.py

@@ -1,6 +1,8 @@
 # Authors: Bastien Trotobas <[email protected]>
 # License: BSD 3-Clause
 
+from __future__ import division, print_function  # Python 2 compatibility
+
 from numpy import ones, inf, sqrt, array, identity, zeros, dot, trace, einsum, absolute
 from numpy.random import seed, choice, randn
 from numpy.testing import run_module_suite, assert_raises, assert_equal, assert_allclose, \
@@ -174,7 +176,7 @@ def test_k_factor(self):
                         err_msg='Wrong k-factor with rician fading in SISO channels')
 
 
-class MIMOTestCase:
+class MIMOTestCase(object):
     msg_length = 100000
     real_mods = array((-1, 1)), array((-3, 3))
     all_mods = array((-1, 1)), array((-3, 3)), \
@@ -218,6 +220,9 @@ def test_SISO(self):
 
 
 class TestMIMODefaultArgs(MIMOTestCase):
+    def __init__(self):
+        super(TestMIMODefaultArgs, self).__init__()
+
     def do(self, nb_tx, nb_rx):
         def check(chan):
             assert_equal(chan.noises, None,
@@ -242,6 +247,9 @@ def check(chan):
 
 @dec.slow
 class TestMIMOFading(MIMOTestCase):
+    def __init__(self):
+        super(TestMIMOFading, self).__init__()
+
     def do(self, nb_tx, nb_rx):
         def check_chan_gain(mod, chan):
             msg = choice(mod, self.msg_length)
@@ -315,6 +323,9 @@ def check_chan_gain(mod, chan):
 
 @dec.slow
 class TestMIMONoiseGeneration(MIMOTestCase):
+    def __init__(self):
+        super(TestMIMONoiseGeneration, self).__init__()
+
     def do(self, nb_tx, nb_rx):
         def check_noise(mod, chan, corrected_SNR_lin):
             msg = choice(mod, self.msg_length)
@@ -348,13 +359,19 @@ def check_noise(mod, chan, corrected_SNR_lin):
 
 
 class TestMIMOTypeCheck(MIMOTestCase):
+    def __init__(self):
+        super(TestMIMOTypeCheck, self).__init__()
+
     def do(self, nb_tx, nb_rx):
         chan = MIMOFlatChannel(nb_tx, nb_rx, 0)
         with assert_raises(TypeError):
             chan.propagate(array((1, 1j)))
 
 
 class TestMIMOShapes(MIMOTestCase):
+    def __init__(self):
+        super(TestMIMOShapes, self).__init__()
+
     def do(self, nb_tx, nb_rx):
         # Without padding
         chan = MIMOFlatChannel(nb_tx, nb_rx, 0)
@@ -382,6 +399,9 @@ def do(self, nb_tx, nb_rx):
 
 
 class TestMIMOkFactor(MIMOTestCase):
+    def __init__(self):
+        super(TestMIMOkFactor, self).__init__()
+
     def do(self, nb_tx, nb_rx):
         prod_nb = nb_tx * nb_rx