view test_framer.py @ 12:9973b7cd0d31 tip

Implement rest of tempo estimator
author Chris Cannam
date Mon, 08 Oct 2012 15:46:54 +0100
parents 7fc62809da94
children
line wrap: on
line source

import framer as fr
import numpy as np

def test_get_frame_count():
    assert fr.get_frame_count(253929, 512) == 496
    assert fr.get_frame_count(0, 59) == 0
    assert fr.get_frame_count(4, 2) == 2
    assert fr.get_frame_count(5, 2) == 3
    assert fr.get_frame_count(6, 1) == 6

def test_get_frame():
    assert (fr.get_frame(np.array([1,2,3,4]), 2, 0) == np.array([1,2])).all()
    assert (fr.get_frame(np.array([1,2,3,4,5]), 2, 2) == np.array([5,0])).all()
    assert (fr.get_frame(np.array([1,2,3,4,5,6]), 1, 3) == np.array([4])).all()