import pytest
from poliastro.twobody.mean_elements import get_mean_elements

def test_get_mean_elements_raises_error_if_invalid_body():
    body = "Sun"

    with pytest.raises(ValueError) as excinfo:
        get_mean_elements(body)
    assert f"The input body is invalid." in excinfo.exconly()

import pytest

from poliastro.twobody.mean_elements import get_mean_elements


def test_get_mean_elements_raises_error_if_invalid_body():
    body = "Sun"

    with pytest.raises(ValueError) as excinfo:
        get_mean_elements(body)
    assert f"The input body '{body}' is invalid." in excinfo.exconly()

Add test for error check
Add test for error check

  import pytest
+ 
  from poliastro.twobody.mean_elements import get_mean_elements
+ 
  
  def test_get_mean_elements_raises_error_if_invalid_body():
      body = "Sun"
  
      with pytest.raises(ValueError) as excinfo:
          get_mean_elements(body)
-     assert f"The input body is invalid." in excinfo.exconly()
+     assert f"The input body '{body}' is invalid." in excinfo.exconly()
  
Add test for error check
## Code Before:
import pytest
from poliastro.twobody.mean_elements import get_mean_elements

def test_get_mean_elements_raises_error_if_invalid_body():
    body = "Sun"

    with pytest.raises(ValueError) as excinfo:
        get_mean_elements(body)
    assert f"The input body is invalid." in excinfo.exconly()

## Instruction:
Add test for error check
## Code After:
import pytest

from poliastro.twobody.mean_elements import get_mean_elements


def test_get_mean_elements_raises_error_if_invalid_body():
    body = "Sun"

    with pytest.raises(ValueError) as excinfo:
        get_mean_elements(body)
    assert f"The input body '{body}' is invalid." in excinfo.exconly()
