154 lines
3.4 KiB
Python
154 lines
3.4 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from nose.tools import raises, assert_items_equal, eq_
|
|
|
|
from forge import get_files_and_intervals, get_timefile_exact, round_timefile,\
|
|
get_timefile
|
|
|
|
eight = datetime(2014, 5, 30, 20)
|
|
nine = datetime(2014, 5, 30, 21)
|
|
ten = datetime(2014, 5, 30, 22)
|
|
|
|
|
|
def minutes(n):
|
|
return timedelta(minutes=n)
|
|
|
|
|
|
def seconds(n):
|
|
return timedelta(seconds=n)
|
|
|
|
## timefile
|
|
|
|
|
|
def test_timefile_exact():
|
|
eq_(get_timefile_exact(eight),
|
|
'2014-05/30/rec-2014-05-30-20-00-00-ror.mp3')
|
|
|
|
## Rounding
|
|
|
|
|
|
def test_rounding_similarity():
|
|
eq_(round_timefile(eight), round_timefile(eight+minutes(20)))
|
|
assert round_timefile(eight) != round_timefile(nine)
|
|
|
|
|
|
def test_rounding_value():
|
|
eq_(round_timefile(eight), eight)
|
|
eq_(round_timefile(eight + minutes(20)), eight)
|
|
|
|
|
|
## Rounding + timefile
|
|
|
|
|
|
def test_timefile_alreadyround():
|
|
eq_(get_timefile(eight),
|
|
'2014-05/30/rec-2014-05-30-20-00-00-ror.mp3')
|
|
|
|
|
|
def test_timefile_toround():
|
|
eq_(get_timefile(eight + minutes(20)),
|
|
'2014-05/30/rec-2014-05-30-20-00-00-ror.mp3')
|
|
|
|
## Intervals
|
|
|
|
|
|
@raises(ValueError)
|
|
def test_intervals_same():
|
|
tuple(get_files_and_intervals(eight, eight))
|
|
|
|
|
|
@raises(ValueError)
|
|
def test_intervals_before():
|
|
tuple(get_files_and_intervals(nine, eight))
|
|
|
|
|
|
def test_intervals_full_1():
|
|
res = list(get_files_and_intervals(eight, nine-seconds(1)))
|
|
eq_(len(res), 1)
|
|
eq_(res[0][1], 0)
|
|
eq_(res[0][2], 0)
|
|
|
|
|
|
def test_intervals_partial_1():
|
|
res = list(get_files_and_intervals(eight, nine-minutes(10)))
|
|
eq_(len(res), 1)
|
|
eq_(res[0][1], 0)
|
|
eq_(res[0][2], 10*60 - 1)
|
|
|
|
|
|
def test_intervals_exact_2():
|
|
res = list(get_files_and_intervals(eight, nine))
|
|
eq_(len(res), 2)
|
|
eq_(res[0][1], 0)
|
|
eq_(res[0][2], 0)
|
|
eq_(res[1][1], 0)
|
|
eq_(res[1][2], 3599)
|
|
|
|
|
|
def test_intervals_partial_2():
|
|
res = list(get_files_and_intervals(eight, nine + minutes(50)))
|
|
eq_(len(res), 2)
|
|
eq_(res[0][1], 0)
|
|
eq_(res[0][2], 0)
|
|
eq_(res[1][1], 0)
|
|
eq_(res[1][2], 599)
|
|
|
|
|
|
def test_intervals_full_2():
|
|
res = list(get_files_and_intervals(eight,
|
|
nine + minutes(59) + seconds(59)))
|
|
eq_(len(res), 2)
|
|
eq_(res[0][1], 0)
|
|
eq_(res[0][2], 0)
|
|
eq_(res[1][1], 0)
|
|
eq_(res[1][2], 0)
|
|
|
|
|
|
def test_intervals_exact_3():
|
|
res = list(get_files_and_intervals(eight, ten))
|
|
eq_(len(res), 3)
|
|
eq_(res[0][1], 0)
|
|
eq_(res[0][2], 0)
|
|
eq_(res[1][1], 0)
|
|
eq_(res[1][2], 0)
|
|
eq_(res[2][1], 0)
|
|
eq_(res[2][2], 3599)
|
|
|
|
|
|
def test_intervals_partial_3():
|
|
res = list(get_files_and_intervals(eight, ten+minutes(50)))
|
|
eq_(len(res), 3)
|
|
eq_(res[0][1], 0)
|
|
eq_(res[0][2], 0)
|
|
eq_(res[1][1], 0)
|
|
eq_(res[1][2], 0)
|
|
eq_(res[2][1], 0)
|
|
eq_(res[2][2], 599)
|
|
|
|
|
|
def test_intervals_full_3():
|
|
res = list(get_files_and_intervals(eight, ten+minutes(59) + seconds(59)))
|
|
eq_(len(res), 3)
|
|
eq_(res[0][1], 0)
|
|
eq_(res[0][2], 0)
|
|
eq_(res[1][1], 0)
|
|
eq_(res[1][2], 0)
|
|
eq_(res[2][1], 0)
|
|
eq_(res[2][2], 0)
|
|
|
|
|
|
def test_intervals_middle_1():
|
|
res = list(get_files_and_intervals(eight + minutes(20),
|
|
nine - minutes(20)))
|
|
eq_(len(res), 1)
|
|
eq_(res[0][1], 20*60)
|
|
eq_(res[0][2], 20*60-1)
|
|
|
|
|
|
def test_intervals_left_2():
|
|
res = list(get_files_and_intervals(eight+minutes(30), nine))
|
|
eq_(len(res), 2)
|
|
eq_(res[0][1], 30*60)
|
|
eq_(res[0][2], 0)
|
|
eq_(res[1][1], 0)
|
|
eq_(res[1][2], 3599)
|