No need for a new class. Just add a new method to the existing class and mover `set_calc_mode()` into the test method.

Also, for consistency this could just go into the `test_write_calc_pr.py` testsuite.