Yes, you can mark tests with the pytest.mark decorator.
Example:
def unit_test_1():
# assert here
def unit_test_2():
# assert here
@pytest.mark.integtest
def integration_test():
# assert here
Now, from the command line, you can run pytest -m "not integtest" for only the unit tests, pytest -m integtest for only the integration test and plain pytest for all.
(You can also decorate your unit tests with pytest.mark.unit if you want, but I find that slightly tedious/verbose)
See the documentation for more information.