def test_register_decoder(): @codecs.decodes('text/plain') def decoder(stream): return Relation( Schema([dict(name="line", type="STRING")]), ( (line.rstrip(),) for line in stream ) ) stream = StringIO("blah\nfoo,1\nbaz") relation = codecs.relation_from(stream, mime_type='text/plain') eq_( relation.schema.fields, [Field(name='line', type='STRING')] ) eq_( list(relation), [ ('blah',), ('foo,1',), ('baz',) ] )
def test_register_decoder(): @codecs.decodes('text/plain') def decoder(stream): return ((line.rstrip(), ) for line in stream) stream = StringIO(u"blah\nfoo,1\nbaz") relation = codecs.relation_from(stream, mime_type='text/plain') eq_(list(relation), [('blah', ), ('foo,1', ), ('baz', )])
def test_decode_csv_relations(): stream = StringIO(u"field1,field2,field3\nfoo,1,0\nbaz,2,0") relation = codecs.relation_from(stream, mime_type='text/csv') assert_sequence_equal( list(relation), [ ['foo','1','0'], ['baz','2','0'] ] )
def test_decode_csv(): stream = StringIO("field1,field2,field3\nfoo,1,0\nbaz,2,0") relation = codecs.relation_from(stream, mime_type='text/csv') eq_( relation.schema.fields, [ Field(name='field1', type='STRING'), Field(name='field2', type='STRING'), Field(name='field3', type='STRING') ] ) assert_sequence_equal( list(relation), [ ['foo','1','0'], ['baz','2','0'] ] )
def test_register_decoder(): @codecs.decodes('text/plain') def decoder(stream): return ( (line.rstrip(),) for line in stream ) stream = StringIO("blah\nfoo,1\nbaz") relation = codecs.relation_from(stream, mime_type='text/plain') eq_( list(relation), [ ('blah',), ('foo,1',), ('baz',) ] )
def test_decode_csv(): stream = StringIO("field1,field2,field3\nfoo,1,0\nbaz,2,0") schema = codecs.schema_from(stream, mime_type='text/csv') eq_( schema, Schema([ Field(name='field1', type='STRING'), Field(name='field2', type='STRING'), Field(name='field3', type='STRING') ]) ) relation = codecs.relation_from(stream, mime_type='text/csv') assert_sequence_equal( list(relation), [ ['foo','1','0'], ['baz','2','0'] ] )
def test_decode_csv_relations(): stream = StringIO(u"field1,field2,field3\nfoo,1,0\nbaz,2,0") relation = codecs.relation_from(stream, mime_type='text/csv') assert_sequence_equal(list(relation), [['foo', '1', '0'], ['baz', '2', '0']])