Cleanup pytest imports to reduce tests boilerplate

[omry]

switching the codebase from using

import pytest

to

from pytest import mark, param, raises, warns

To eliminate usage of pytest. everywhere in tests, making lines shorter and code formatting cleaner.

[omry]

cc @odelalleau ,@jieru:
May likely cause conflicts for tests in outstanding diffs.