Searching in test cases/python