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()