Add missing numpy mock