Skip to content

Commit 5fd39ea

Browse files
committed
TST: Fix existing tests since Period and int are comparable
1 parent c7a140e commit 5fd39ea

File tree

1 file changed

+4
-15
lines changed

1 file changed

+4
-15
lines changed

pandas/tests/scalar/test_period.py

Lines changed: 4 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -942,47 +942,36 @@ def test_notEqual(self):
942942

943943
def test_greater(self):
944944
assert self.february > self.january1
945+
assert self.february > self.january1.ordinal
945946

946947
def test_greater_Raises_Value(self):
947948
with pytest.raises(period.IncompatibleFrequency):
948949
self.january1 > self.day
949950

950-
def test_greater_Raises_Type(self):
951-
with pytest.raises(TypeError):
952-
self.january1 > 1
953-
954951
def test_greaterEqual(self):
955952
assert self.january1 >= self.january2
953+
assert self.january1 >= self.january2.ordinal
956954

957955
def test_greaterEqual_Raises_Value(self):
958956
with pytest.raises(period.IncompatibleFrequency):
959957
self.january1 >= self.day
960958

961-
with pytest.raises(TypeError):
962-
print(self.january1 >= 1)
963-
964959
def test_smallerEqual(self):
965960
assert self.january1 <= self.january2
961+
assert self.january1 <= self.january2.ordinal
966962

967963
def test_smallerEqual_Raises_Value(self):
968964
with pytest.raises(period.IncompatibleFrequency):
969965
self.january1 <= self.day
970966

971-
def test_smallerEqual_Raises_Type(self):
972-
with pytest.raises(TypeError):
973-
self.january1 <= 1
974-
975967
def test_smaller(self):
976968
assert self.january1 < self.february
969+
assert self.january1 < self.february.ordinal
977970

978971
def test_smaller_Raises_Value(self):
979972
with pytest.raises(period.IncompatibleFrequency):
980973
self.january1 < self.day
981974

982-
def test_smaller_Raises_Type(self):
983-
with pytest.raises(TypeError):
984-
self.january1 < 1
985-
986975
def test_sort(self):
987976
periods = [self.march, self.january1, self.february]
988977
correctPeriods = [self.january1, self.february, self.march]

0 commit comments

Comments
 (0)