Skip to content

Commit c30c862

Browse files
committed
Moving tests
1 parent 5849db5 commit c30c862

File tree

3 files changed

+40
-35
lines changed

3 files changed

+40
-35
lines changed

pandas/tests/indexes/common.py

+38
Original file line numberDiff line numberDiff line change
@@ -852,3 +852,41 @@ def test_fillna(self):
852852
expected[1] = True
853853
self.assert_numpy_array_equal(idx._isnan, expected)
854854
self.assertTrue(idx.hasnans)
855+
856+
def test_map(self):
857+
for name, index in self.indices.items():
858+
if len(index) == 0 or isinstance(index, MultiIndex):
859+
pass
860+
else:
861+
# Applying a function to the Index
862+
index = index.map(lambda x: x)
863+
print(name, index)
864+
self.assertTrue(isinstance(index, Index))
865+
#self.assertTrue(index.equals(Index(['I1', 'I2'])))
866+
#self.assertEqual(index.name, "Numbering")
867+
#
868+
#testIdx = self.unicodeIndex.map(lambda x: len(x))
869+
#tm.assert_index_equal(testIdx, Int64Index([10]*100))
870+
#
871+
#testIdx = self.strIndex.map(lambda x: len(x))
872+
#tm.assert_index_equal(testIdx, Int64Index([10]*100))
873+
#
874+
#testIdx = self.dateIndex.map(lambda x: x + timedelta(days=1))
875+
#tm.assert_index_equal(
876+
# testIdx, DatetimeIndex([dt + timedelta(days=1) for dt in tm.makeDateIndex(100)]))
877+
#
878+
#testIdx = self.periodIndex.map(lambda x: x.to_timestamp())
879+
#tm.assert_index_equal(testIdx, self.dateIndex)
880+
#
881+
#testIdx = self.intIndex.map(lambda x: str(x))
882+
#tm.assert_index_equal(testIdx, Index([str(i) for i in range(100)]))
883+
#
884+
#testIdx = self.floatIndex.map(lambda x: -1 if x < 0 else 1)
885+
#self.assertEqual(len(testIdx), 100)
886+
#self.assertTrue(isinstance(testIdx, Int64Index))
887+
#self.assertTrue(set(testIdx == {-1, 1}))
888+
#
889+
#testIdx = self.boolIndex.map(lambda x: not x)
890+
#tm.assert_index_equal(testIdx, Index([False, True]))
891+
#
892+
#testIdx = self.catIndex.map(lambda x: len(x))

pandas/tests/indexes/test_base.py

-35
Original file line numberDiff line numberDiff line change
@@ -1744,41 +1744,6 @@ def test_string_index_repr(self):
17441744

17451745
self.assertEqual(coerce(idx), expected)
17461746

1747-
def test_map(self):
1748-
1749-
# Applying a function to the Index
1750-
df = pd.DataFrame([[0, 1], [2, 3]], columns=['c1', 'c2'], index=['i1', 'i2'])
1751-
df.index.name = "Numbering"
1752-
df.index = df.index.map(lambda x: x.upper())
1753-
self.assertTrue(df.index.equals(Index(['I1', 'I2'])))
1754-
self.assertEqual(df.index.name, "Numbering")
1755-
1756-
testIdx = self.unicodeIndex.map(lambda x: len(x))
1757-
tm.assert_index_equal(testIdx, Int64Index([10]*100))
1758-
1759-
testIdx = self.strIndex.map(lambda x: len(x))
1760-
tm.assert_index_equal(testIdx, Int64Index([10]*100))
1761-
1762-
testIdx = self.dateIndex.map(lambda x: x + timedelta(days=1))
1763-
tm.assert_index_equal(
1764-
testIdx, DatetimeIndex([dt + timedelta(days=1) for dt in tm.makeDateIndex(100)]))
1765-
1766-
testIdx = self.periodIndex.map(lambda x: x.to_timestamp())
1767-
tm.assert_index_equal(testIdx, self.dateIndex)
1768-
1769-
testIdx = self.intIndex.map(lambda x: str(x))
1770-
tm.assert_index_equal(testIdx, Index([str(i) for i in range(100)]))
1771-
1772-
testIdx = self.floatIndex.map(lambda x: -1 if x < 0 else 1)
1773-
self.assertEqual(len(testIdx), 100)
1774-
self.assertTrue(isinstance(testIdx, Int64Index))
1775-
self.assertTrue(set(testIdx == {-1, 1}))
1776-
1777-
testIdx = self.boolIndex.map(lambda x: not x)
1778-
tm.assert_index_equal(testIdx, Index([False, True]))
1779-
1780-
testIdx = self.catIndex.map(lambda x: len(x))
1781-
17821747

17831748
class TestMixedIntIndex(Base, tm.TestCase):
17841749
# Mostly the tests from common.py for which the results differ

pandas/tests/indexes/test_datetimelike.py

+2
Original file line numberDiff line numberDiff line change
@@ -1149,3 +1149,5 @@ def test_fillna_timedelta(self):
11491149
exp = pd.Index(
11501150
[pd.Timedelta('1 day'), 'x', pd.Timedelta('3 day')], dtype=object)
11511151
self.assert_index_equal(idx.fillna('x'), exp)
1152+
1153+

0 commit comments

Comments
 (0)