ABX discrimination task in python