add test .py from thomas