Skip to content

CLN: Parameterize test case #22327

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.

Sign up for GitHub

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
Aug 14, 2018
Merged

CLN: Parameterize test case #22327

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
63abad5
Parameterize test case
alimcmaster1 Aug 13, 2018
ea6f018
flake8 add new line
alimcmaster1 Aug 13, 2018
cfff07c
Combine similar test functions
alimcmaster1 Aug 13, 2018
cde2500
Fix test name typo
alimcmaster1 Aug 13, 2018
b78995c
Update arg names
alimcmaster1 Aug 14, 2018
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
43 changes: 13 additions & 30 deletions pandas/tests/scalar/timestamp/test_unary_ops.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,26 +19,19 @@ class TestTimestampUnaryOps(object):

# --------------------------------------------------------------
# Timestamp.round

def test_round_day_naive(self):
dt = Timestamp('20130101 09:10:11')
result = dt.round('D')
expected = Timestamp('20130101')
assert result == expected

dt = Timestamp('20130101 19:10:11')
result = dt.round('D')
expected = Timestamp('20130102')
assert result == expected

dt = Timestamp('20130201 12:00:00')
result = dt.round('D')
expected = Timestamp('20130202')
assert result == expected

dt = Timestamp('20130104 12:00:00')
result = dt.round('D')
expected = Timestamp('20130105')
@pytest.mark.parametrize('timestamp, freq, expected', [
('20130101 09:10:11', 'D', '20130101'),
('20130101 19:10:11', 'D', '20130102'),
('20130201 12:00:00', 'D', '20130202'),
('20130104 12:00:00', 'D', '20130105'),
('2000-01-05 05:09:15.13', 'D', '2000-01-05 00:00:00'),
('2000-01-05 05:09:15.13', 'H', '2000-01-05 05:00:00'),
('2000-01-05 05:09:15.13', 'S', '2000-01-05 05:09:15')
])
def test_round_frequencies(self, timestamp, freq, expected):
dt = Timestamp(timestamp)
result = dt.round(freq)
expected = Timestamp(expected)
assert result == expected

def test_round_tzaware(self):
Expand Down Expand Up @@ -85,16 +78,6 @@ def test_round_invalid_arg(self):
with tm.assert_raises_regex(ValueError, INVALID_FREQ_ERR_MSG):
stamp.round('foo')

@pytest.mark.parametrize('freq, expected', [
('D', Timestamp('2000-01-05 00:00:00')),
('H', Timestamp('2000-01-05 05:00:00')),
('S', Timestamp('2000-01-05 05:09:15'))])
def test_round_frequencies(self, freq, expected):
stamp = Timestamp('2000-01-05 05:09:15.13')

result = stamp.round(freq=freq)
assert result == expected

@pytest.mark.parametrize('test_input, rounder, freq, expected', [
('2117-01-01 00:00:45', 'floor', '15s', '2117-01-01 00:00:45'),
('2117-01-01 00:00:45', 'ceil', '15s', '2117-01-01 00:00:45'),
Expand Down