# $Id$

import sys

from writer import *

try:
    import xml.etree.cElementTree as ET
except ImportError:
    import ElementTree as ET

args = sys.argv[1:]
if not args:
    print "usage: testwriter.py (echo|traverse|traverse-time|writer)*"

if not args or "echo" in args:
    print "simple.xml:"
    write(ET.parse("simple.xml"), sys.stdout.write)
    print
    print "simple-ns.xml:"
    write(ET.parse("simple-ns.xml"), sys.stdout.write)
    print
    print "simple-ns.xml:"
    write(ET.parse("simple-ns.xml"), sys.stdout.write, default_namespace="namespace")
    print

if "traverse" in args:

    tree = ET.parse("simple.xml").getroot()
    for item in traverse(tree):
        print item

if "traverse-time" in args:

    tree = ET.parse("hamlet.xml").getroot()

    import time
    t0 = time.clock()
    for item in traverse(tree):
        pass
    print time.clock() - t0

if "writer" in args:

    tree = ET.parse("hamlet.xml")
    # tree = ET.parse("simple-ns.xml")

    # write(tree, sys.stdout.write, namespace_map=[("", "namespace")])

    from cStringIO import StringIO

    import time
    t0 = time.clock()
    for i in range(10):
        write(tree, StringIO().write, namespace_map=[("", "namespace")])
    print time.clock() - t0

    t0 = time.clock()
    for i in range(10):
        tree.write(StringIO())
    print time.clock() - t0