Use test data for python notebook