Add tests from python