Python shape_match Examples

Example #1

File: test_feature.py Project: keunwoochoi/pumpp

def test_feature_stft(audio, SR, HOP_LENGTH, n_fft):

    ext = pumpp.feature.STFT(name='stft',
                             sr=SR, hop_length=HOP_LENGTH,
                             n_fft=n_fft)

    output = ext.transform(**audio)

    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #2

File: test_feature.py Project: keunwoochoi/pumpp

def test_feature_mel(audio, SR, HOP_LENGTH, n_fft, n_mels):

    ext = pumpp.feature.Mel(name='mel',
                            sr=SR, hop_length=HOP_LENGTH,
                            n_fft=n_fft, n_mels=n_mels)

    output = ext.transform(**audio)

    # Check the fields
    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #3

File: test_feature.py Project: keunwoochoi/pumpp

def test_feature_cqtphasediff(audio, SR, HOP_LENGTH, over_sample, n_octaves):

    ext = pumpp.feature.CQTPhaseDiff(name='cqt',
                                     sr=SR, hop_length=HOP_LENGTH,
                                     n_octaves=n_octaves,
                                     over_sample=over_sample)

    output = ext.transform(**audio)

    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #4

def test_feature_stft(audio, SR, HOP_LENGTH, n_fft):

    ext = pumpp.feature.STFT(name='stft',
                             sr=SR,
                             hop_length=HOP_LENGTH,
                             n_fft=n_fft)

    output = ext.transform(**audio)

    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #5

def test_feature_cqtphasediff(audio, SR, HOP_LENGTH, over_sample, n_octaves):

    ext = pumpp.feature.CQTPhaseDiff(name='cqt',
                                     sr=SR,
                                     hop_length=HOP_LENGTH,
                                     n_octaves=n_octaves,
                                     over_sample=over_sample)

    output = ext.transform(**audio)

    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #6

File: test_feature.py Project: tomxi/pumpp

def test_feature_time(audio, SR, HOP_LENGTH, conv, dtype):

    ext = pumpp.feature.TimePosition(name='time',
                                     sr=SR,
                                     hop_length=HOP_LENGTH,
                                     conv=conv,
                                     dtype=dtype)

    output = ext.transform(**audio)

    # Check the fields
    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #7

File: test_feature.py Project: tomxi/pumpp

def test_feature_tempogram(audio, SR, HOP_LENGTH, WIN_LENGTH, conv, dtype):

    ext = pumpp.feature.Tempogram(name='rhythm',
                                  sr=SR,
                                  hop_length=HOP_LENGTH,
                                  win_length=WIN_LENGTH,
                                  conv=conv,
                                  dtype=dtype)

    output = ext.transform(**audio)

    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #8

def test_feature_mel(audio, SR, HOP_LENGTH, n_fft, n_mels):

    ext = pumpp.feature.Mel(name='mel',
                            sr=SR,
                            hop_length=HOP_LENGTH,
                            n_fft=n_fft,
                            n_mels=n_mels)

    output = ext.transform(**audio)

    # Check the fields
    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #9

File: test_feature.py Project: tomxi/pumpp

def test_feature_stft_mag(audio, SR, HOP_LENGTH, n_fft, conv, log, dtype):

    ext = pumpp.feature.STFTMag(name='stft',
                                sr=SR,
                                hop_length=HOP_LENGTH,
                                n_fft=n_fft,
                                conv=conv,
                                log=log,
                                dtype=dtype)

    output = ext.transform(**audio)

    # Check the fields
    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)

Example #10

File: test_feature.py Project: tomxi/pumpp

def test_feature_hcqtmag(audio, SR, HOP_LENGTH, over_sample, n_octaves, hconv,
                         log, harmonics, dtype):

    ext = pumpp.feature.HCQTMag(name='hcqt',
                                sr=SR,
                                hop_length=HOP_LENGTH,
                                n_octaves=n_octaves,
                                over_sample=over_sample,
                                conv=hconv,
                                log=log,
                                harmonics=harmonics,
                                dtype=dtype)

    output = ext.transform(**audio)

    assert set(output.keys()) == set(ext.fields.keys())

    for key in ext.fields:
        assert shape_match(output[key].shape[1:], ext.fields[key].shape)
        assert type_match(output[key].dtype, ext.fields[key].dtype)