Converting From unittest to pytest#

Conversion Checklist#

Note

Please bear in mind the following checklist is for general use; there may be some cases which require extra context or thought before implementing these changes.

  1. Before making any manual changes, run dannysepler/pytestify on the file. This does a lot of the brunt work for you!

  2. Check for references to IrisTest. If a class inherits from this, remove the inheritance. Inheritance is unnecessary for pytest tests, so iris.tests.IrisTest has been deprecated and its convenience methods have been moved to the iris.tests._shared_utils module.

  3. Check for references to unittest. Many of the functions within unittest are also in pytest, so often you can just change where the function is imported from.

  4. Check for references to self.assert. Pytest has a lighter-weight syntax for assertions, e.g. assert x == 2 instead of assertEqual(x, 2). In the case of custom IrisTest assertions, the majority of these have been replicated in iris.tests._shared_utils, but with snake_case instead of camelCase. Some iris.tests.IrisTest assertions have not been converted into iris.tests._shared_utils, as these were deemed easy to achieve via simple assert ... statements.

  5. Check for references to setup_method(). Replace this with _setup() instead. Ensure that this is decorated with @pytest.fixture(autouse=True).

    @pytest.fixture(autouse=True)
    def _setup(self):
       ...
    
  6. Check for references to super(). Most test classes used to inherit from iris.tests.IrisTest, so references to this should be removed. Any inheritance patterns in _setup() (see above) can be achieved by ‘chaining’ fixtures; note that each setup fixture needs a unique name:

    class TestFoo:
        @pytest.fixture(autouse=True)
        def _setup_foo(self):
            ...
    
    class TestBar(TestFoo):
        @pytest.fixture(autouse=True)
        def _setup(self, _setup_foo):
            ...
    
  7. Check for references to @tests. These should be changed to @_shared_utils.

  8. Check for mock.patch("warnings.warn"). This can be replaced with pytest.warns(match=message).

  9. Check for references to mock or self.patch. These should be changed to use the mocker fixture - see the pytest-mock docs. Note that pytest-mock’s patch does not support the context-manager syntax; in most cases this is made unnecessary (see Usage as context manager), in advanced cases consider using the monkeypatch fixture to provide a context-manager.

  10. Check for np.testing.assert.... This can usually be swapped for _shared_utils.assert....

  11. Check for np.allclose. This should be swapped for _shared_utils.assert_array_all_close.

  12. Check for references to self.tmp_dir and self.temp_filename. In pytest, tmp_path is used instead, and can be passed into functions as a fixture.

  13. Check for if __name__ == 'main'. This is no longer needed with pytest.

  14. Remove the top-level import of iris.tests (usually import iris.tests as tests). Having followed the above steps, any remaining calls (e.g. iris.tests.get_data_path()) should be easily replacable with calls to iris.tests._shared_utils (e.g. iris.tests._shared_utils.get_data_path()).

  15. Ensure that all test classes start with Test. Tests will not run in pytest without it.

  16. Check the file against astral-sh/ruff , using pip install ruff -> ruff check --select PT <file>.

  17. Ensure that all the tests are passing. Some tests are set to skip if certain packages aren’t installed in your environment. These are often also skipped in the Iris CI also, so make sure that they run and pass locally.

Common Translations#

unittest method

pytest equivalent

assertTrue(x)

assert x

assertFalse(x)

assert not x

assertRegex(x, y)

assert re.match(y, x)

assertRaisesRegex(cls, msg_re)

with pytest.raises(cls, match=msg_re):

mock.patch(...)

mocker.patch(...)

with mock.patch.object(...) as x:

x = mocker.patch.object(...)