Moved unit tests. Got State wrapper running in python