assert_raises -> pytest.raises