def setUp(self):
"""Called before each test."""
self.gmdmimo_object = GMDMimo()
class GMDMimoTestCase(unittest.TestCase):
def setUp(self):
"""Called before each test."""
self.gmdmimo_object = GMDMimo()
def test_encode(self):
# xxxxxxxxxx test the case with Ntx=2 xxxxxxxxxxxxxxxxxxxxxxxxxxxxx
Nt = 2
Nr = 2
data = np.r_[0:15*Nt]
channel = randn_c(Nr, Nt)
self.gmdmimo_object.set_channel_matrix(channel)
encoded_data = self.gmdmimo_object.encode(data)
# data_aux = data.reshape(Nt, -1)
U, S, V_H = np.linalg.svd(channel)
_, _, P = gmd(U, S, V_H)
W = P / math.sqrt(Nt)
expected_encoded_data = W.dot(data.reshape(Nr, -1))
np.testing.assert_array_almost_equal(
expected_encoded_data, encoded_data)
# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
# xxxxx Test if an exception is raised for wrong size xxxxxxxxxxxxx
# The exception is raised if the input array size is not a multiple
# of the number of transmit antennas
data2 = np.r_[0:15*Nt+1]
with self.assertRaises(ValueError):
self.gmdmimo_object.encode(data2)
# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
def test_decode(self):
# xxxxxxxxxx test the case with Ntx=2, NRx=2 xxxxxxxxxxxxxxxxxxxxxx
Nt = 2
Nr = 2
data = np.r_[0:15*Nt]
channel = randn_c(Nr, Nt)
self.gmdmimo_object.set_channel_matrix(channel)
encoded_data = self.gmdmimo_object.encode(data)
received_data = channel.dot(encoded_data)
decoded_data = self.gmdmimo_object.decode(received_data)
np.testing.assert_array_almost_equal(data, decoded_data)
# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
def test_calc_post_processing_SINRs(self):
Nr = 3
Nt = 3
noise_var = 0.01
channel = randn_c(Nr, Nt)
self.gmdmimo_object.set_noise_var(noise_var)
self.gmdmimo_object.set_channel_matrix(channel)
W = self.gmdmimo_object._calc_precoder(channel)
G_H = self.gmdmimo_object._calc_receive_filter(channel, noise_var)
expected_sinrs = linear2dB(calc_SINRs(channel, W, G_H, noise_var))
# Calculate the SINR using the function in the mimo module. Note
# that we need to pass the channel, the precoder, the receive
# filter and the noise variance.
sinrs = calc_post_processing_SINRs(channel, W, G_H, noise_var)
np.testing.assert_array_almost_equal(sinrs, expected_sinrs, 1)
# Calculate the SINR using method in the MIMO class. Note that we
# only need to pass the noise variance, since the mimo object knows
# the channel and it can calculate the precoder and receive filter.
sinrs_other = self.gmdmimo_object.calc_linear_SINRs(noise_var)
np.testing.assert_array_almost_equal(sinrs_other, expected_sinrs, 1)