import pytest


def test_class():
    """ Test HeartData

    """
    from heartdata import HeartData
    # Check if data splits into time, voltage
    test1 = HeartData('test_data/test_data1')
    assert test1.time is not None
    assert test1.voltage is not None
    assert test1.time[9] == 0.025
    assert test1.voltage[9] == -0.135
    assert test1.duration == 27.775
    assert test1.num_beats == 35
    assert test1.mean_hr_bpm > 0
    assert test1.mean_hr_bpm < 100
    assert type(test1.voltage_extremes) == tuple

import pytest


def test_class():
    """ Test HeartData

    """
    from heartdata import HeartData
    # Check if data splits into time, voltage
    test1 = HeartData('test_data/test_data1')
    assert test1.time is not None
    assert test1.voltage is not None
    assert test1.time[9] == 0.025
    assert test1.voltage[9] == -0.135
    assert test1.duration == 27.775
    assert test1.num_beats == 35
    assert test1.mean_hr_bpm > 0
    assert test1.mean_hr_bpm < 100
    assert type(test1.voltage_extremes) == tuple

Test HeartData # Check if data splits into time, voltage