Beispiel #1
0
def test_file_notfound(hea_path):
    with pytest.raises(FileNotFoundError):
        ECGRecord.from_wfdb(hea_path)
Beispiel #2
0
def test_select_annotation(hea_path):
    record = ECGRecord.from_wfdb(hea_path)
    assert len(record.annotations.select_label("N")) == 2239
Beispiel #3
0
def test_unique_labels(hea_path):
    record = ECGRecord.from_wfdb(hea_path)
    assert record.annotations.unique_labels == ['+', 'A', 'N', 'V']
Beispiel #4
0
def test_signal_content(hea_path, lead_name, signal):
    record = ECGRecord.from_wfdb(hea_path)
    assert record.get_lead(lead_name)[:10] == signal
Beispiel #5
0
def test_annotation(hea_path):
    record = ECGRecord.from_wfdb(hea_path)
    assert len(record.annotations) == 2274
Beispiel #6
0
def test_signal_name(hea_path):
    record = ECGRecord.from_wfdb(hea_path)
    assert record.lead_names == ['MLII', 'V5']
Beispiel #7
0
def test_record_name(hea_path):
    record = ECGRecord.from_wfdb(hea_path)
    assert record.record_name == "100"
Beispiel #8
0
def test_signal(hea_path):
    record = ECGRecord.from_wfdb(hea_path)
    assert record.n_sig == 2
Beispiel #9
0
def test_length(hea_path):
    record = ECGRecord.from_wfdb(hea_path)
    assert len(record) == 650000
Beispiel #10
0
with zipfile.ZipFile("MIT_Noise.zip", "r") as zip_ref:
    zip_ref.extractall()

hea_path_1 = "mit-bih-arrhythmia-database-1.0.0/mit-bih-arrhythmia-database-1.0.0/"
hea_path = "mit-bih-noise-stress-test-database-1.0.0/mit-bih-noise-stress-test-database-1.0.0/bw.hea"

# To load a wfdb formatted ECG record
hea_path_1 = "D:/Research_Impl/Noise_Removal_Autoencoder/Dataset/mit-bih-arrhythmia-database-1.0.0/mit-bih-arrhythmia-database-1.0.0/"
hea_path = "D:/Research_Impl/Noise_Removal_Autoencoder/Dataset/mit-bih-noise-stress-test-database-1.0.0 (1)/mit-bih-noise-stress-test-database-1.0.0/bw.hea"

#######################################################################################################
#                                      EXTRACTING THE NOISE SIGNALS FROM HEA
#######################################################################################################

record = ECGRecord.from_wfdb(hea_path)
signal_noise = record.get_lead("noise1")

#######################################################################################################
#                                      EXTRACTING THE ORIGINAL SIGNALS FROM HEA
#######################################################################################################
signal_original = []
for i in range(100, 125):
    if i == 110 or i == 120:
        continue
    record_1 = ECGRecord.from_wfdb(hea_path_1 + str(i) + ".hea")
    signal_orig = record_1.get_lead("MLII")
    if signal_orig == None:
        print(i)
        continue
    signal_original.append(signal_orig)