def test_axioms():
cg = Test_Chunk_Generator()
assert len(Chunk(cg.empty_chunk)) == 0
assert not Chunk(cg.empty_chunk)
assert len(Chunk(cg.chunks_list[0])) > 0
assert Chunk(cg.chunks_list[0])
test_chunk = \
{ 7: {'source': 'foo', 'data': numpy.array([2,4,8])},
8: {'source': 'foo', 'data': numpy.array([1,2,3])},
9: {'source': 'foo', 'data': numpy.array([[1,2],[1,2],[1,2]])},
10: {'source': 'foo', 'data': numpy.array(['str1','str2','str3'])}}
assert chunks.is_equal(test_chunk, cg.chunks_list[0])
assert Chunk(test_chunk) == cg.chunks_list[0]
assert not chunks.is_equal(test_chunk, cg.chunks_list[1])
assert Chunk(test_chunk) != cg.chunks_list[1]
def test_average_generator():
avg_chunks = [{ 7: {'source': 'foo', 'data': numpy.array([9,14])},
8: {'source': 'foo', 'data': numpy.array([5.5, numpy.NAN])},
9: {'source': 'foo', 'data': numpy.array([[1,2],[-9223372036854775808,-9223372036854775808]])},
10: {'source': 'foo', 'data': numpy.array(['str1','nan'])}},
{ 7: {'source': 'foo', 'data': numpy.array([15])},
8: {'source': 'foo', 'data': numpy.array([8])},
9: {'source': 'foo', 'data': numpy.array([[1,2]])},
10: {'source': 'foo', 'data': numpy.array(['str2'])}}]
ag = Average_Generator(Test_Chunk_Generator())
i = 0
for chunk in ag.chunks(Test_StreamRequest([9,14,15,20])):
assert chunks.is_equal(chunk, avg_chunks[i], exact=False)
i+=1
def test_interpolation_generator():
int_chunks = [{ 7: {'source': 'foo', 'data': numpy.array([1,5])},
8: {'source': 'foo', 'data': numpy.array([1,2.25])},
9: {'source': 'foo', 'data': numpy.array([[1,2],[1,2]])},
10: {'source': 'foo', 'data': numpy.array(['str1','str2'])}},
{ 7: {'source': 'foo', 'data': numpy.array([10,15])},
8: {'source': 'foo', 'data': numpy.array([4.75,(7. + (2./6.))])},
9: {'source': 'foo', 'data': numpy.array([[1,2],[1,2]])},
10: {'source': 'foo', 'data': numpy.array(['str1','str1'])}},
{ 7: {'source': 'foo', 'data': numpy.array([20])},
8: {'source': 'foo', 'data': numpy.array([8])},
9: {'source': 'foo', 'data': numpy.array([[1,2]])},
10: {'source': 'foo', 'data': numpy.array(['str2'])}}]
ag = Interpolation_Generator(Test_Chunk_Generator())
i = 0
for chunk in ag.chunks(Test_StreamRequest([1,5,10,15,20])):
assert chunks.is_equal(chunk, int_chunks[i], exact=False)
i+=1
