Add more test cases for python blocks