def test_QAM_bb_qpsk_src(self): np.random.seed(100) x_test, b_test, t_test = (np.array([ 0.00585723 + 0.00585723j, -0.00275016 - 0.00275016j, -0.00164540 - 0.01335987j, 0.00887646 + 0.01437677j, -0.01540288 + 0.01131686j, 0.00480440 - 0.02394915j, 0.02505607 + 0.02585128j, -0.04406383 - 0.00716616j, -0.02797722 - 0.08626139j, 0.11024504 + 0.1600832j, 0.00580570 + 0.16357483j, -0.44629859 - 0.76924864j, -0.96387506 - 1.22739262j, -1.32990076 + 0.11435352j, -1.06357060 + 1.2446656j, 0.04406383 + 0.22076409j, 1.06649175 - 1.18402117j, 1.21485132 - 1.24832839j, 0.97347224 - 0.94165619j, 0.88372072 - 0.89875699j ]), np.array([ -0.00293625, 0.00137866, 0.00376109, -0.00582846, 0.00102417, 0.00479866, -0.01276, 0.01284087, 0.02863407, -0.06775815, -0.04245547, 0.30467864, 0.54924435, 0.30467864, -0.04245547, -0.06775815, 0.02863407, 0.01284087, -0.01276, 0.00479866, 0.00102417, -0.00582846, 0.00376109, 0.00137866, -0.00293625 ]), np.array([ -1. - 1.j, -1. + 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, -1. + 1.j, -1. - 1.j, -1. - 1.j, -1. + 1.j ])) x, b, t = dc.QAM_bb(10, 2, mod_type='qpsk', pulse='src') npt.assert_almost_equal(x, x_test) npt.assert_almost_equal(b, b_test) npt.assert_almost_equal(t, t_test)
def test_OFDM_rx_channel_estimate(self): z_out_test, H_out_test = ( np.array( [ -2.91356233 - 0.93854058j, -3.03083561 - 1.01177886j, 3.10687062 + 1.09962706j, 2.91679784 + 2.79392693j, 2.95621370 + 0.87789714j, 2.93521287 + 1.12869418j, -3.17675560 - 1.0834705j, 1.25700626 + 1.19497994j, 1.16433902 + 2.62068101j, -3.10408334 + 1.08514004j, 1.02623864 + 3.01672402j, -0.98366297 - 1.21602375j, 0.89577012 + 1.07687508j, 1.05852406 + 1.05134363j, 0.93287609 - 3.11042385j, 0.99965390 + 0.88124784j, -1.16293758 + 3.08562314j, -2.84891079 - 1.07199168j, -3.22236927 - 2.90425199j, 3.07028549 - 2.88413491j, -3.12192058 + 2.89625467j, 3.18017151 - 1.09375776j, -2.78212772 + 3.05087219j, 1.13471595 - 2.89218144j, -3.17092453 - 1.11298847j, 3.10927184 + 0.86801524j, -0.76520964 - 3.32101721j, -0.94935570 + 2.86081052j, 0.93535950 + 1.10545223j, 1.09394518 - 1.17966519j, 3.10748055 + 1.12377382j, -3.12337017 - 0.89848715j, -2.95725651 + 0.97491592j, 3.14041238 - 3.01998896j, -1.05440640 + 3.04843936j, -0.94130790 - 0.82179287j, -0.79049810 - 1.04083796j, 2.96004080 + 1.01692442j, -3.13063510 + 1.32083138j, -2.58084447 - 3.28171534j, 3.09664605 + 0.82140179j, 2.87565015 - 1.17002378j, 2.82351021 + 2.83242155j, 2.99238994 + 3.06883778j, -0.83601519 - 2.8886988j, 3.05383614 + 1.22402533j, -0.92550302 + 0.92366226j, -0.97707573 + 3.08608891j, 0.73489228 - 2.99163649j, 2.89544691 + 2.76671634j ]), np.array( [ 1.49261307 - 0.12886832j, 1.36399692 - 0.24831791j, 1.24438887 - 0.41524198j, 1.15276504 - 0.47480443j, 1.09981815 - 0.35438673j, 0.86684483 - 0.31710329j, 0.75885865 - 0.23542562j, 0.76309583 - 0.19374055j, 0.61556098 + 0.09731796j, 0.77281595 + 0.07096727j, 0.87593303 + 0.15642133j, 1.06728467 + 0.29788462j, 1.08613086 + 0.23650714j, 1.12082635 + 0.09129381j, 1.31026672 + 0.17419224j, 1.19459330 + 0.01027668j, 1.19745209 + 0.11471611j, 1.36689249 - 0.07997548j, 1.26471663 - 0.07505238j, 1.14356226 - 0.19961235j, 0.84149706 - 0.21609579j, 0.85489994 - 0.18101042j, 0.79502365 - 0.17155484j, 0.71666634 - 0.02650505j, 0.82384118 + 0.0565963j, 0.74313589 + 0.28403893j, 0.88570493 + 0.29345603j, 0.95203301 + 0.37888469j, 0.98676887 + 0.4108844j, 1.26869289 + 0.35672436j, 1.44594176 + 0.3296819j, 1.48817425 + 0.07577518j ])) hc = np.array([1.0, 0.1, -0.05, 0.15, 0.2, 0.05]) x1, b1, IQ_data1 = dc.QAM_bb(50000, 1, '16qam') x_out = dc.OFDM_tx(IQ_data1, 32, 64, 100, True, 10) c_out = signal.lfilter(hc, 1, x_out) # Apply channel distortion r_out = dc.cpx_AWGN(c_out, 25, 64 / 32) # Es/N0 = 25 dB z_out, H = dc.OFDM_rx(r_out, 32, 64, 100, True, 10, alpha=0.95, ht=hc) npt.assert_almost_equal(z_out[:50], z_out_test) npt.assert_almost_equal(H[:50], H_out_test)
def test_ODFM_tx(self): x_out_test = np.array([ 0.00000000 + 0.125j, -0.10185331 + 0.27369942j, -0.10291586 + 0.12529202j, -0.05485981 - 0.1015143j, -0.02143872 - 0.09787268j, -0.06906044 + 0.05231368j, -0.18815224 + 0.050888j, -0.26164122 - 0.15836327j, -0.21940048 - 0.36048543j, -0.14486054 - 0.38169759j, -0.11830476 - 0.25561157j, -0.07250935 - 0.12760226j, 0.05301567 - 0.08413918j, 0.14316564 - 0.07020723j, 0.07590886 + 0.01736066j, -0.04551924 + 0.15686941j, -0.03125000 + 0.21875j, 0.09755018 + 0.17168517j, 0.15431728 + 0.10974492j, 0.08889087 + 0.04259743j, 0.04284671 - 0.1107734j, 0.10071736 - 0.25986197j, 0.15582045 - 0.17226253j, 0.06652251 + 0.12312402j, -0.15245874 + 0.29798543j, -0.32346606 + 0.23845079j, -0.25311017 + 0.21460293j, 0.07831717 + 0.3396657j, 0.43085592 + 0.30360811j, 0.48116320 - 0.0505655j, 0.16656460 - 0.32765262j, -0.20071609 - 0.16142259j, -0.25000000 + 0.1875j, 0.04290155 + 0.25900306j, 0.33313987 + 0.08484705j, 0.28478134 - 0.00986648j, -0.05936711 + 0.00190181j, -0.30195965 - 0.0628197j, -0.12280721 - 0.1651266j, 0.31807654 - 0.16252886j, 0.53190048 - 0.13951457j, 0.31342228 - 0.20065005j, 0.00806130 - 0.17969398j, -0.00105255 + 0.03378639j, 0.15279016 + 0.16494501j, 0.09844557 - 0.009236j, -0.11589986 - 0.20597693j, -0.10438721 - 0.09983656j, 0.15625000 + 0.09375j, 0.22805837 + 0.03951473j, ]) x1, b1, IQ_data1 = dc.QAM_bb(50000, 1, '16qam') x_out = dc.OFDM_tx(IQ_data1, 32, 64, 0, True, 0) npt.assert_almost_equal(x_out[:50], x_out_test)
def test_QAM_SEP_256qam(self): Nsymb_test, Nerr_test, SEP_test = (4986, 2190, 0.43922984356197353) x, b, tx_data = dc.QAM_bb(5000, 10, '256qam', 'src') x = dc.cpx_AWGN(x, 20, 10) y = signal.lfilter(b, 1, x) Nsymb, Nerr, SEP = dc.QAM_SEP(tx_data, y[10 + 10 * 12::10], '256qam', Ntransient=0) self.assertEqual(Nsymb, Nsymb_test) self.assertEqual(Nerr, Nerr_test) self.assertEqual(SEP, SEP_test)
def test_QAM_SEP_qpsk(self): Nsymb_test, Nerr_test, SEP_test = (4986, 0, 0.0) x, b, tx_data = dc.QAM_bb(5000, 10, 'qpsk', 'src') x = dc.cpx_AWGN(x, 20, 10) y = signal.lfilter(b, 1, x) Nsymb, Nerr, SEP = dc.QAM_SEP(tx_data, y[10 + 10 * 12::10], 'qpsk', Ntransient=0) self.assertEqual(Nsymb, Nsymb_test) self.assertEqual(Nerr, Nerr_test) self.assertEqual(SEP, SEP_test)
def test_QAM_SEP_64qam(self): Nsymb_test, Nerr_test, SEP_test = (4986, 245, 0.04913758523866827) x, b, tx_data = dc.QAM_bb(5000, 10, '64qam', 'src') x = dc.cpx_AWGN(x, 20, 10) y = signal.lfilter(b, 1, x) Nsymb, Nerr, SEP = dc.QAM_SEP(tx_data, y[10 + 10 * 12::10], '64qam', Ntransient=0) self.assertEqual(Nsymb, Nsymb_test) self.assertEqual(Nerr, Nerr_test) self.assertEqual(SEP, SEP_test)
def test_QAM_SEP_16qam_error(self): Nsymb_test, Nerr_test, SEP_test = (9976, 172, 0.017241379310344827) x, b, tx_data = dc.QAM_bb(10000, 1, '16qam', 'rect') x = dc.cpx_AWGN(x, 15, 1) y = signal.lfilter(b, 1, x) Nsymb, Nerr, SEP = dc.QAM_SEP(tx_data, y[1 * 12::1], '16qam', Ntransient=0) self.assertEqual(Nsymb, Nsymb_test) self.assertEqual(Nerr, Nerr_test) self.assertEqual(SEP, SEP_test)
def test_QAM_bb_qpsk_rect(self): x_test, b_test, t_test = (np.array([ -1. - 1.j, -1. - 1.j, -1. + 1.j, -1. + 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, -1. + 1.j, -1. + 1.j, -1. - 1.j, -1. - 1.j, -1. - 1.j, -1. - 1.j, -1. + 1.j, -1. + 1.j ]), np.array([0.5, 0.5]), np.array([ -1. - 1.j, -1. + 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, -1. + 1.j, -1. - 1.j, -1. - 1.j, -1. + 1.j ])) x, b, t = dc.QAM_bb(10, 2, mod_type='qpsk', pulse='rect') npt.assert_almost_equal(x, x_test) npt.assert_almost_equal(b, b_test) npt.assert_almost_equal(t, t_test)
def test_OFDM_rx(self): z_out_test, H_test = ( np.array( [ -3.11740028 - 0.90748269j, -3.11628187 - 0.88948888j, 2.88565859 + 1.13255112j, 2.89076997 + 3.16052588j, 2.90396853 + 1.19595053j, 2.93439648 + 1.23703401j, -3.00724063 - 0.72880083j, 1.07519281 + 1.27075039j, 1.14472192 + 3.22099905j, -2.82962216 + 1.15148633j, 1.16245397 + 3.09533441j, -0.85799363 - 0.94063529j, 1.12036257 + 1.03825793j, 1.10109739 + 1.02622557j, 1.08488052 - 2.98041713j, 1.07132873 + 1.01625511j, -0.92119499 + 3.01872286j, -2.91683903 - 0.9906338j, -2.91213253 - 3.00295552j, 3.09229992 - 3.01974828j ]), np.array([ 1.42289223 - 1.43696423e-01j, 1.34580486 - 2.66705232e-01j, 1.23071709 - 3.51736667e-01j, 1.09530096 - 3.87688911e-01j, 0.95992898 - 3.71339473e-01j, 0.84428862 - 3.07656711e-01j, 0.76421410 - 2.08710707e-01j, 0.72928932 - 9.14213562e-02j, 0.74161551 + 2.54124114e-02j, 0.79590656 + 1.24244087e-01j, 0.88082345 + 1.91510354e-01j, 0.98123573 + 2.19504072e-01j, 1.08094630 + 2.07118520e-01j, 1.16536231 + 1.59418930e-01j, 1.22364402 + 8.62177170e-02j, 1.25000000 - 2.77555756e-17j, 1.25000000 + 2.77555756e-17j, 1.22364402 - 8.62177170e-02j, 1.16536231 - 1.59418930e-01j, 1.08094630 - 2.07118520e-01j, 0.98123573 - 2.19504072e-01j, 0.88082345 - 1.91510354e-01j, 0.79590656 - 1.24244087e-01j, 0.74161551 - 2.54124114e-02j, 0.72928932 + 9.14213562e-02j, 0.76421410 + 2.08710707e-01j, 0.84428862 + 3.07656711e-01j, 0.95992898 + 3.71339473e-01j, 1.09530096 + 3.87688911e-01j, 1.23071709 + 3.51736667e-01j, 1.34580486 + 2.66705232e-01j, 1.42289223 + 1.43696423e-01j ])) hc = np.array([1.0, 0.1, -0.05, 0.15, 0.2, 0.05]) x1, b1, IQ_data1 = dc.QAM_bb(50000, 1, '16qam') x_out = dc.OFDM_tx(IQ_data1, 32, 64, 0, True, 0) c_out = signal.lfilter(hc, 1, x_out) # Apply channel distortion r_out = dc.cpx_AWGN(c_out, 100, 64 / 32) # Es/N0 = 100 dB z_out, H = dc.OFDM_rx(r_out, 32, 64, -1, True, 0, alpha=0.95, ht=hc) npt.assert_almost_equal(z_out[:20], z_out_test) npt.assert_almost_equal(H, H_test)
def test_QAM_bb_64qam_rect(self): x_test, b_test, t_test = (np.array([ -1.00000000 - 0.42857143j, -1.00000000 - 0.42857143j, -1.00000000 + 0.42857143j, -1.00000000 + 0.42857143j, -0.14285714 - 0.42857143j, -0.14285714 - 0.42857143j, 1.00000000 - 0.42857143j, 1.00000000 - 0.42857143j, 1.00000000 + 0.71428571j, 1.00000000 + 0.71428571j, 1.00000000 - 0.42857143j, 1.00000000 - 0.42857143j, -1.00000000 - 0.71428571j, -1.00000000 - 0.71428571j, -0.42857143 - 1.j, -0.42857143 - 1.j, 0.71428571 - 1.j, 0.71428571 - 1.j, 0.14285714 + 1.j, 0.14285714 + 1.j ]), np.array([0.5, 0.5]), np.array([ -7. - 3.j, -7. + 3.j, -1. - 3.j, 7. - 3.j, 7. + 5.j, 7. - 3.j, -7. - 5.j, -3. - 7.j, 5. - 7.j, 1. + 7.j ])) x, b, t = dc.QAM_bb(10, 2, mod_type='64qam', pulse='rect') npt.assert_almost_equal(x, x_test) npt.assert_almost_equal(b, b_test) npt.assert_almost_equal(t, t_test)
def test_QAM_bb_16qam_rect(self): x_test, b_test, t_test = (np.array([ -1.00000000 + 0.33333333j, -1.00000000 + 0.33333333j, -1.00000000 - 0.33333333j, -1.00000000 - 0.33333333j, 1.00000000 + 0.33333333j, 1.00000000 + 0.33333333j, 1.00000000 + 0.33333333j, 1.00000000 + 0.33333333j, 1.00000000 + 0.33333333j, 1.00000000 + 0.33333333j, 1.00000000 + 0.33333333j, 1.00000000 + 0.33333333j, -1.00000000 - 0.33333333j, -1.00000000 - 0.33333333j, 0.33333333 - 1.j, 0.33333333 - 1.j, 0.33333333 - 1.j, 0.33333333 - 1.j, -1.00000000 + 1.j, -1.00000000 + 1.j ]), np.array([0.5, 0.5]), np.array([ -3. + 1.j, -3. - 1.j, 3. + 1.j, 3. + 1.j, 3. + 1.j, 3. + 1.j, -3. - 1.j, 1. - 3.j, 1. - 3.j, -3. + 3.j ])) x, b, t = dc.QAM_bb(10, 2, mod_type='16qam', pulse='rect') npt.assert_almost_equal(x, x_test) npt.assert_almost_equal(b, b_test) npt.assert_almost_equal(t, t_test)
def test_QAM_bb_qpsk_rc(self): x_test, b_test, t_test = (np.array([ -2.22799382e-18 - 2.22799382e-18j, -4.07129297e-03 - 4.07129297e-03j, 2.22160609e-19 + 4.67814826e-18j, -2.22059175e-03 + 5.92199418e-03j, 6.43926133e-18 - 2.91703518e-18j, 1.97462099e-02 + 7.90222150e-03j, -9.75189466e-18 - 1.28297996e-17j, -4.09888874e-02 - 7.30785022e-02j, 1.05934616e-17 + 3.71417032e-17j, 9.37971907e-02 + 2.31071828e-01j, -1.52549076e-18 - 6.78758025e-17j, -4.10719565e-01 - 8.26448725e-01j, -1.00000000e+00 - 1.00000000e+00j, -1.36231531e+00 + 1.25147025e-01j, -1.00000000e+00 + 1.00000000e+00j, 4.09888874e-02 + 2.75737546e-01j, 1.00000000e+00 - 1.00000000e+00j, 1.24877191e+00 - 1.36728049e+00j, 1.00000000e+00 - 1.00000000e+00j, 8.23659721e-01 - 7.64661456e-01j ]), np.array([ 1.11223990e-18, 2.03243583e-03, -1.22314501e-18, -9.23891131e-04, -8.79167710e-19, -6.90120596e-03, 5.63651951e-18, 2.84718702e-02, -1.19149691e-17, -8.10891577e-02, 1.73229581e-17, 3.08804252e-01, 4.99211393e-01, 3.08804252e-01, 1.73229581e-17, -8.10891577e-02, -1.19149691e-17, 2.84718702e-02, 5.63651951e-18, -6.90120596e-03, -8.79167710e-19, -9.23891131e-04, -1.22314501e-18, 2.03243583e-03, 1.11223990e-18 ]), np.array([ -1. - 1.j, -1. + 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, 1. - 1.j, -1. + 1.j, -1. - 1.j, -1. - 1.j, -1. + 1.j ])) x, b, t = dc.QAM_bb(10, 2, mod_type='qpsk', pulse='rc') npt.assert_almost_equal(x, x_test) npt.assert_almost_equal(b, b_test) npt.assert_almost_equal(t, t_test)
def test_QAM_bb_256qam_rect(self): x_test, b_test, t_test = (np.array([ 0.06666667 - 0.73333333j, 0.06666667 - 0.73333333j, 0.06666667 - 0.33333333j, 0.06666667 - 0.33333333j, -0.60000000 - 0.73333333j, -0.60000000 - 0.73333333j, -0.06666667 - 0.73333333j, -0.06666667 - 0.73333333j, -0.06666667 + 0.86666667j, -0.06666667 + 0.86666667j, 1.00000000 - 0.73333333j, 1.00000000 - 0.73333333j, -1.00000000 - 0.86666667j, -1.00000000 - 0.86666667j, 0.33333333 - 1.j, 0.33333333 - 1.j, 0.86666667 + 0.06666667j, 0.86666667 + 0.06666667j, -0.46666667 + 1.j, -0.46666667 + 1.j ]), np.array([0.5, 0.5]), np.array([ 1. - 11.j, 1. - 5.j, -9. - 11.j, -1. - 11.j, -1. + 13.j, 15. - 11.j, -15. - 13.j, 5. - 15.j, 13. + 1.j, -7. + 15.j ])) x, b, t = dc.QAM_bb(10, 2, mod_type='256qam', pulse='rect') npt.assert_almost_equal(x, x_test) npt.assert_almost_equal(b, b_test) npt.assert_almost_equal(t, t_test)
1 / np.linalg.norm(symbols_comparison[0, 0]) #0.707 #varianza np.var(symbols_comparison) #2 #the sign of a number. np.sign([-5., 4.5]) np.sign(0) #Test Array testarray = np.array([]) testarray = (2, 2) #Ejemplo Ofdm x1, b1, IQ_data1 = dc.QAM_bb(50000, 1, '16qam') x_out = dc.OFDM_tx(IQ_data1, 32, 64) plt.psd(x_out, 2**10, 1) plt.xlabel(r'Normalized Frequency ($\omega/(2\pi)=f/f_s$)') plt.ylim([-40, 0]) plt.xlim([-.5, .5]) #Round elements of the array to the nearest integer. a = np.array([-1.7, -1.5, -0.2, 0.2, 1.5, 1.7, 2.0]) np.rint(a) #Draw samples from a binomial distribution. np.random.binomial(size=3, n=1, p=0.5) #size: number of experiments; n: number of trails; p: probability of success # Agregar elementos Array
def test_QAM_bb_mod_error(self): with self.assertRaisesRegexp(ValueError, 'Unknown mod_type'): x, b, t = dc.QAM_bb(10, 2, mod_type='unknown')
def test_QAM_bb_pulse_error(self): with self.assertRaisesRegexp(ValueError, 'pulse shape must be src, rc, or rect'): dc.QAM_bb(10, 2, pulse='value')