Skip to content

TST: add back test_generic.py, accid removed in the big reorg #18114

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Nov 7, 2017
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 12 additions & 10 deletions pandas/core/generic.py
Original file line number Diff line number Diff line change
Expand Up @@ -352,7 +352,7 @@ def _get_axis_number(self, axis):
else:
try:
return self._AXIS_NUMBERS[axis]
except:
except KeyError:
pass
raise ValueError('No axis named {0} for object type {1}'
.format(axis, type(self)))
Expand All @@ -365,7 +365,7 @@ def _get_axis_name(self, axis):
else:
try:
return self._AXIS_NAMES[axis]
except:
except KeyError:
pass
raise ValueError('No axis named {0} for object type {1}'
.format(axis, type(self)))
Expand Down Expand Up @@ -701,7 +701,7 @@ def squeeze(self, axis=None):
return self.iloc[
tuple([0 if i in axis and len(a) == 1 else slice(None)
for i, a in enumerate(self.axes)])]
except:
except Exception:
return self

def swaplevel(self, i=-2, j=-1, axis=0):
Expand Down Expand Up @@ -1021,7 +1021,7 @@ def __invert__(self):
try:
arr = operator.inv(_values_from_object(self))
return self.__array_wrap__(arr)
except:
except Exception:

# inv fails with 0 len
if not np.prod(self.shape):
Expand Down Expand Up @@ -1907,7 +1907,7 @@ def _maybe_update_cacher(self, clear=False, verify_is_copy=True):
else:
try:
ref._maybe_cache_changed(cacher[0], self)
except:
except Exception:
pass

if verify_is_copy:
Expand Down Expand Up @@ -2016,15 +2016,15 @@ def _check_setitem_copy(self, stacklevel=4, t='setting', force=False):
if not gc.get_referents(self.is_copy()):
self.is_copy = None
return
except:
except Exception:
pass

# we might be a false positive
try:
if self.is_copy().shape == self.shape:
self.is_copy = None
return
except:
except Exception:
pass

# a custom message
Expand Down Expand Up @@ -2999,7 +2999,7 @@ def reindex(self, *args, **kwargs):
if self._needs_reindex_multi(axes, method, level):
try:
return self._reindex_multi(axes, copy, fill_value)
except:
except Exception:
pass

# perform the reindex on the axes
Expand Down Expand Up @@ -3715,7 +3715,7 @@ def _check_inplace_setting(self, value):
try:
if np.isnan(value):
return True
except:
except Exception:
pass

raise TypeError('Cannot do inplace boolean setting on '
Expand Down Expand Up @@ -5005,6 +5005,8 @@ def clip(self, lower=None, upper=None, axis=None, inplace=False,
inplace = validate_bool_kwarg(inplace, 'inplace')

axis = nv.validate_clip_with_axis(axis, args, kwargs)
if axis is not None:
axis = self._get_axis_number(axis)

# GH 17276
# numpy doesn't like NaN as a clip value
Expand Down Expand Up @@ -5916,7 +5918,7 @@ def _where(self, cond, other=np.nan, inplace=False, axis=None, level=None,
new_other = _values_from_object(self).copy()
new_other[icond] = other
other = new_other
except:
except Exception:
try_quick = False

# let's create a new (if we failed at the above
Expand Down
18 changes: 18 additions & 0 deletions pandas/tests/frame/test_analytics.py
Original file line number Diff line number Diff line change
Expand Up @@ -1822,6 +1822,24 @@ def test_built_in_round(self):
{'col1': [1., 2., 3.], 'col2': [1., 2., 3.]})
tm.assert_frame_equal(round(df), expected_rounded)

def test_pct_change(self):
# GH 11150
pnl = DataFrame([np.arange(0, 40, 10), np.arange(0, 40, 10), np.arange(
0, 40, 10)]).astype(np.float64)
pnl.iat[1, 0] = np.nan
pnl.iat[1, 1] = np.nan
pnl.iat[2, 3] = 60

mask = pnl.isnull()

for axis in range(2):
expected = pnl.ffill(axis=axis) / pnl.ffill(axis=axis).shift(
axis=axis) - 1
expected[mask] = np.nan
result = pnl.pct_change(axis=axis, fill_method='pad')

tm.assert_frame_equal(result, expected)

# Clip

def test_clip(self):
Expand Down
82 changes: 79 additions & 3 deletions pandas/tests/frame/test_timeseries.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,12 +11,15 @@
import numpy as np

from pandas import (DataFrame, Series, Index,
Timestamp, DatetimeIndex,
to_datetime, date_range)
Timestamp, DatetimeIndex, MultiIndex,
to_datetime, date_range, period_range)
import pandas as pd
import pandas.tseries.offsets as offsets

from pandas.util.testing import assert_series_equal, assert_frame_equal
from pandas.util.testing import (assert_series_equal,
assert_frame_equal,
assert_index_equal,
assert_raises_regex)

import pandas.util.testing as tm
from pandas.compat import product
Expand Down Expand Up @@ -601,3 +604,76 @@ def test_frame_to_period(self):
tm.assert_index_equal(pts.columns, exp.columns.asfreq('M'))

pytest.raises(ValueError, df.to_period, axis=2)

@pytest.mark.parametrize("fn", ['tz_localize', 'tz_convert'])
def test_tz_convert_and_localize(self, fn):
l0 = date_range('20140701', periods=5, freq='D')

# TODO: l1 should be a PeriodIndex for testing
# after GH2106 is addressed
with pytest.raises(NotImplementedError):
period_range('20140701', periods=1).tz_convert('UTC')
with pytest.raises(NotImplementedError):
period_range('20140701', periods=1).tz_localize('UTC')
# l1 = period_range('20140701', periods=5, freq='D')
l1 = date_range('20140701', periods=5, freq='D')

int_idx = Index(range(5))

if fn == 'tz_convert':
l0 = l0.tz_localize('UTC')
l1 = l1.tz_localize('UTC')

for idx in [l0, l1]:

l0_expected = getattr(idx, fn)('US/Pacific')
l1_expected = getattr(idx, fn)('US/Pacific')

df1 = DataFrame(np.ones(5), index=l0)
df1 = getattr(df1, fn)('US/Pacific')
assert_index_equal(df1.index, l0_expected)

# MultiIndex
# GH7846
df2 = DataFrame(np.ones(5), MultiIndex.from_arrays([l0, l1]))

df3 = getattr(df2, fn)('US/Pacific', level=0)
assert not df3.index.levels[0].equals(l0)
assert_index_equal(df3.index.levels[0], l0_expected)
assert_index_equal(df3.index.levels[1], l1)
assert not df3.index.levels[1].equals(l1_expected)

df3 = getattr(df2, fn)('US/Pacific', level=1)
assert_index_equal(df3.index.levels[0], l0)
assert not df3.index.levels[0].equals(l0_expected)
assert_index_equal(df3.index.levels[1], l1_expected)
assert not df3.index.levels[1].equals(l1)

df4 = DataFrame(np.ones(5),
MultiIndex.from_arrays([int_idx, l0]))

# TODO: untested
df5 = getattr(df4, fn)('US/Pacific', level=1) # noqa

assert_index_equal(df3.index.levels[0], l0)
assert not df3.index.levels[0].equals(l0_expected)
assert_index_equal(df3.index.levels[1], l1_expected)
assert not df3.index.levels[1].equals(l1)

# Bad Inputs

# Not DatetimeIndex / PeriodIndex
with assert_raises_regex(TypeError, 'DatetimeIndex'):
df = DataFrame(index=int_idx)
df = getattr(df, fn)('US/Pacific')

# Not DatetimeIndex / PeriodIndex
with assert_raises_regex(TypeError, 'DatetimeIndex'):
df = DataFrame(np.ones(5),
MultiIndex.from_arrays([int_idx, l0]))
df = getattr(df, fn)('US/Pacific', level=0)

# Invalid level
with assert_raises_regex(ValueError, 'not valid'):
df = DataFrame(index=l0)
df = getattr(df, fn)('US/Pacific', level=1)
Empty file.
Loading