from __future__ import absolute_import, division, print_function

import pytest
from astropy import units as u

from astrodynamics.util import verify_unit


def test_verify_unit():
    # Implicit dimensionless values are allowed, test that Quantity is returned.
    assert verify_unit(0, u.one) == 0 * u.one

    # Test failure mode
    with pytest.raises(ValueError):
        verify_unit(0, u.meter)

    # Quantity should be passed back if unit matches
    assert verify_unit(1 * u.meter, u.meter) == 1 * u.meter

from __future__ import absolute_import, division, print_function

import pytest
from astropy import units as u

from astrodynamics.util import verify_unit


def test_verify_unit():
    # Implicit dimensionless values are allowed, test that Quantity is returned.
    assert verify_unit(0, u.one) == 0 * u.one
    assert verify_unit(0, '') == 0 * u.one

    # Test failure mode
    with pytest.raises(ValueError):
        verify_unit(0, u.meter)
    with pytest.raises(ValueError):
        verify_unit(0, 'm')

    # Quantity should be passed back if unit matches
    assert verify_unit(1 * u.meter, u.meter) == 1 * u.meter
    assert verify_unit(1 * u.meter, 'm') == 1 * u.meter

Test string form of verify_unit
Test string form of verify_unit

  from __future__ import absolute_import, division, print_function
  
  import pytest
  from astropy import units as u
  
  from astrodynamics.util import verify_unit
  
  
  def test_verify_unit():
      # Implicit dimensionless values are allowed, test that Quantity is returned.
      assert verify_unit(0, u.one) == 0 * u.one
+     assert verify_unit(0, '') == 0 * u.one
  
      # Test failure mode
      with pytest.raises(ValueError):
          verify_unit(0, u.meter)
+     with pytest.raises(ValueError):
+         verify_unit(0, 'm')
  
      # Quantity should be passed back if unit matches
      assert verify_unit(1 * u.meter, u.meter) == 1 * u.meter
+     assert verify_unit(1 * u.meter, 'm') == 1 * u.meter
  
Test string form of verify_unit
## Code Before:
from __future__ import absolute_import, division, print_function

import pytest
from astropy import units as u

from astrodynamics.util import verify_unit


def test_verify_unit():
    # Implicit dimensionless values are allowed, test that Quantity is returned.
    assert verify_unit(0, u.one) == 0 * u.one

    # Test failure mode
    with pytest.raises(ValueError):
        verify_unit(0, u.meter)

    # Quantity should be passed back if unit matches
    assert verify_unit(1 * u.meter, u.meter) == 1 * u.meter

## Instruction:
Test string form of verify_unit
## Code After:
from __future__ import absolute_import, division, print_function

import pytest
from astropy import units as u

from astrodynamics.util import verify_unit


def test_verify_unit():
    # Implicit dimensionless values are allowed, test that Quantity is returned.
    assert verify_unit(0, u.one) == 0 * u.one
    assert verify_unit(0, '') == 0 * u.one

    # Test failure mode
    with pytest.raises(ValueError):
        verify_unit(0, u.meter)
    with pytest.raises(ValueError):
        verify_unit(0, 'm')

    # Quantity should be passed back if unit matches
    assert verify_unit(1 * u.meter, u.meter) == 1 * u.meter
    assert verify_unit(1 * u.meter, 'm') == 1 * u.meter
