-
Notifications
You must be signed in to change notification settings - Fork 13
/
test.py
32 lines (28 loc) · 948 Bytes
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from polarcodes.PolarCode import PolarCode
import numpy as np
import json
# simulate test code
myPC = PolarCode(64, 32)
myPC.simulate(save_to='data/pc_sim', Eb_No_vec=np.arange(1,5), design_SNR=5.0, manual_const_flag=False)
myPC.plot(['pc_sim'], 'data/')
# read FER and BER data from test code
with open('data/pc_sim.json') as json_file:
data = json.load(json_file)
FER_data = np.array(data['FER'])
BER_data = np.array(data['BER'])
# compare test code with known correct values
FER_test_data = np.array([
0.313,
0.126,
0.03,
0.004125412541254125
])
BER_test_data = np.array([
0.09709375,
0.03740625,
0.00815625,
0.0010184612211221122
])
check_cond = np.logical_and(np.abs(FER_test_data-FER_data) < 1e-3, np.abs(BER_test_data-BER_data) < 1e-3)
if np.sum(check_cond) == check_cond.shape[0]:
print("The library is ready to go!")