def test_add_and_get():
ss = SongSegment()
ss.add(1, 0, 5000, None, None, None, None)
seg = ss.get(1)
assert seg['song_id'] == 1
assert seg['time_from'] == 0
assert seg['time_to'] == 5000
assert seg['mfcc'] is None
assert seg['chroma'] is None
assert seg['tempogram'] is None
assert seg['similar'] is None
def test_get_all():
ss = SongSegment()
ss.add(1, 0, 5000, None, None, None, None)
seg = ss.get_all()
assert len(seg) > 0