Add tests for `prime_factor_multiplicity()`