@@ -13000,7 +13000,7 @@ def test_select_dtypes_include(self):
13000
13000
df = DataFrame({'a': list('abc'),
13001
13001
'b': list(range(1, 4)),
13002
13002
'c': np.arange(3, 6).astype('u1'),
13003
- 'd': np.arange(4.0, 7.0),
13003
+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
13004
13004
'e': [True, False, True]})
13005
13005
ri = df.select_dtypes(include=[np.number])
13006
13006
ei = df[['b', 'c', 'd']]
@@ -13010,7 +13010,7 @@ def test_select_dtypes_exclude(self):
13010
13010
df = DataFrame({'a': list('abc'),
13011
13011
'b': list(range(1, 4)),
13012
13012
'c': np.arange(3, 6).astype('u1'),
13013
- 'd': np.arange(4.0, 7.0),
13013
+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
13014
13014
'e': [True, False, True]})
13015
13015
re = df.select_dtypes(exclude=[np.number])
13016
13016
ee = df[['a', 'e']]
@@ -13020,7 +13020,7 @@ def test_select_dtypes_exclude_include(self):
13020
13020
df = DataFrame({'a': list('abc'),
13021
13021
'b': list(range(1, 4)),
13022
13022
'c': np.arange(3, 6).astype('u1'),
13023
- 'd': np.arange(4.0, 7.0),
13023
+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
13024
13024
'e': [True, False, True],
13025
13025
'f': pd.date_range('now', periods=3).values})
13026
13026
exclude = np.datetime64,
@@ -13030,7 +13030,7 @@ def test_select_dtypes_exclude_include(self):
13030
13030
tm.assert_frame_equal(r, e)
13031
13031
13032
13032
exclude = 'datetime',
13033
- include = 'bool', 'int '
13033
+ include = 'bool', 'int64', 'int32 '
13034
13034
r = df.select_dtypes(include=include, exclude=exclude)
13035
13035
e = df[['b', 'e']]
13036
13036
tm.assert_frame_equal(r, e)
@@ -13039,7 +13039,7 @@ def test_select_dtypes_not_an_attr_but_still_valid_dtype(self):
13039
13039
df = DataFrame({'a': list('abc'),
13040
13040
'b': list(range(1, 4)),
13041
13041
'c': np.arange(3, 6).astype('u1'),
13042
- 'd': np.arange(4.0, 7.0),
13042
+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
13043
13043
'e': [True, False, True],
13044
13044
'f': pd.date_range('now', periods=3).values})
13045
13045
df['g'] = df.f.diff()
@@ -13071,7 +13071,7 @@ def test_select_dtypes_bad_datetime64(self):
13071
13071
df = DataFrame({'a': list('abc'),
13072
13072
'b': list(range(1, 4)),
13073
13073
'c': np.arange(3, 6).astype('u1'),
13074
- 'd': np.arange(4.0, 7.0),
13074
+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
13075
13075
'e': [True, False, True],
13076
13076
'f': pd.date_range('now', periods=3).values})
13077
13077
with tm.assertRaisesRegexp(ValueError, '.+ is too specific'):
@@ -13085,7 +13085,7 @@ def test_select_dtypes_str_raises(self):
13085
13085
'g': list(u('abc')),
13086
13086
'b': list(range(1, 4)),
13087
13087
'c': np.arange(3, 6).astype('u1'),
13088
- 'd': np.arange(4.0, 7.0),
13088
+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
13089
13089
'e': [True, False, True],
13090
13090
'f': pd.date_range('now', periods=3).values})
13091
13091
string_dtypes = set((str, 'str', np.string_, 'S1',
@@ -13107,7 +13107,7 @@ def test_select_dtypes_bad_arg_raises(self):
13107
13107
'g': list(u('abc')),
13108
13108
'b': list(range(1, 4)),
13109
13109
'c': np.arange(3, 6).astype('u1'),
13110
- 'd': np.arange(4.0, 7.0),
13110
+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
13111
13111
'e': [True, False, True],
13112
13112
'f': pd.date_range('now', periods=3).values})
13113
13113
with tm.assertRaisesRegexp(TypeError, 'data type.*not understood'):
0 commit comments