Updated test

Author: ryankarlos

pandas/tests/scalar/timedelta/test_timedelta.py

@@ -348,15 +348,19 @@ def test_unit_parser(self, units, np_unit, wrapper):
 
  result = to_timedelta('2{}'.format(unit))
  assert result == expected
- result = Timedelta('2{}'.format(unit))
+ result = Timedelta('2{}'.format(units))
  assert result == expected
 
  @pytest.mark.parametrize('units', ['Y', 'y', 'M'])
- def test_unit_M_Y_deprecated(self, units):
- with tm.assert_produces_warning(FutureWarning):
+ def test_unit_m_y_deprecated(self, units):
+ with pytest.warns(FutureWarning) as record1:
  to_timedelta(10, units)
- TimedeltaIndex([1, 1, 1], units)
+ assert len(record1) == 1
+ assert str(record1[0].message) == "M and Y units are deprecated."
+ with pytest.warns(FutureWarning) as record2:
  Timedelta(10, units)
+ assert len(record2) == 1
+ assert str(record2[0].message) == "M and Y units are deprecated."
 
  def test_numeric_conversions(self):
  assert ct(0) == np.timedelta64(0, 'ns')