| Baptiste Lepilleur | 4cd8bae | 2007-03-15 22:11:38 +0000 | [diff] [blame^] | 1 | # Simple implementation of a json test runner to run the test against json-py.
|
| 2 |
|
| 3 | import sys
|
| 4 | import os.path
|
| 5 | import json
|
| 6 | import types
|
| 7 |
|
| 8 | if len(sys.argv) != 2:
|
| 9 | print "Usage: %s input-json-file", sys.argv[0]
|
| 10 | sys.exit(3)
|
| 11 |
|
| 12 | input_path = sys.argv[1]
|
| 13 | base_path = os.path.splitext(input_path)[0]
|
| 14 | actual_path = base_path + '.actual'
|
| 15 | rewrite_path = base_path + '.rewrite'
|
| 16 | rewrite_actual_path = base_path + '.actual-rewrite'
|
| 17 |
|
| 18 | def valueTreeToString( fout, value, path = '.' ):
|
| 19 | ty = type(value)
|
| 20 | if ty is types.DictType:
|
| 21 | fout.write( '%s={}\n' % path )
|
| 22 | suffix = path[-1] != '.' and '.' or ''
|
| 23 | names = value.keys()
|
| 24 | names.sort()
|
| 25 | for name in names:
|
| 26 | valueTreeToString( fout, value[name], path + suffix + name )
|
| 27 | elif ty is types.ListType:
|
| 28 | fout.write( '%s=[]\n' % path )
|
| 29 | for index, childValue in zip( xrange(0,len(value)), value ):
|
| 30 | valueTreeToString( fout, childValue, path + '[%d]' % index )
|
| 31 | elif ty is types.StringType:
|
| 32 | fout.write( '%s="%s"\n' % (path,value) )
|
| 33 | elif ty is types.IntType:
|
| 34 | fout.write( '%s=%d\n' % (path,value) )
|
| 35 | elif ty is types.FloatType:
|
| 36 | fout.write( '%s=%.16g\n' % (path,value) )
|
| 37 | elif value is True:
|
| 38 | fout.write( '%s=true\n' % path )
|
| 39 | elif value is False:
|
| 40 | fout.write( '%s=false\n' % path )
|
| 41 | elif value is None:
|
| 42 | fout.write( '%s=null\n' % path )
|
| 43 | else:
|
| 44 | assert False and "Unexpected value type"
|
| 45 |
|
| 46 | def parseAndSaveValueTree( input, actual_path ):
|
| 47 | root = json.read( input )
|
| 48 | fout = file( actual_path, 'wt' )
|
| 49 | valueTreeToString( fout, root )
|
| 50 | fout.close()
|
| 51 | return root
|
| 52 |
|
| 53 | def rewriteValueTree( value, rewrite_path ):
|
| 54 | rewrite = json.write( value )
|
| 55 | rewrite = rewrite[1:-1] # Somehow the string is quoted ! jsonpy bug ?
|
| 56 | file( rewrite_path, 'wt').write( rewrite + '\n' )
|
| 57 | return rewrite
|
| 58 |
|
| 59 | input = file( input_path, 'rt' ).read()
|
| 60 | root = parseAndSaveValueTree( input, actual_path )
|
| 61 | rewrite = rewriteValueTree( json.write( root ), rewrite_path )
|
| 62 | rewrite_root = parseAndSaveValueTree( rewrite, rewrite_actual_path )
|
| 63 |
|
| 64 | sys.exit( 0 )
|