Restructed python tests