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