A magnetization tool for post-assembly magnetization of a magnet assembly including a main coil, an end surface of the main coil configured to be positioned substantially parallel to an outer surface of the magnet assembly for magnetizing a magnetic pole of the magnet assembly, the main coil being configured to generate a magnetic field and a shielding arrangement positioned adjacent the main coil in a plane substantially parallel to the end surface of the main coil, whereby the shielding arrangement is configured to generate a shielding magnetic field, whereby a resulting magnetic field of the shielding magnetic field and the magnetic field is substantially only protruding the magnetic pole of the magnet assembly and directly adjacent magnetic poles of the magnet assembly such that the magnetic pole of the magnet assembly and the directly adjacent magnetic poles of the magnetic pole have a substantially opposite polarity.