def test_init(self):
channel1 = randn_c(3)
mrt_object1 = MRT(channel1)
self.assertEqual(3, mrt_object1.Nt)
self.assertEqual(1, mrt_object1.Nr)
channel2 = randn_c(1, 3)
mrt_object2 = MRT(channel2)
self.assertEqual(3, mrt_object2.Nt)
self.assertEqual(1, mrt_object2.Nr)
channel3 = randn_c(2, 3)
# Number of receive antennas must be exact 1. Since channel3 has 2
# receive antennas, an exception should be raised
with self.assertRaises(ValueError):
MRT(channel3)
def setUp(self):
"""Called before each test."""
self.mrt_object = MRT()
