a new test variation for a three label mnist subset