Skip to content

Commit 3b4d839

Browse files
jrebackjreback
committed
Merge pull request #6577 from bwignall/assert_6175_isinstance
CLN: Change assert_([not] isinstance(a,b)) to specialized forms
2 parents 9e12645 + df3eaee commit 3b4d839

15 files changed

+62
-48
lines changed

‎pandas/io/tests/test_packers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -359,7 +359,7 @@ def test_multi(self):
359359
l = [self.frame['float'], self.frame['float']
360360
.A, self.frame['float'].B, None]
361361
l_rec = self.encode_decode(l)
362-
self.assert_(isinstance(l_rec, tuple))
362+
self.assertIsInstance(l_rec, tuple)
363363
check_arbitrary(l, l_rec)
364364

365365
def test_iterator(self):

‎pandas/io/tests/test_parsers.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -415,7 +415,7 @@ def test_multiple_date_cols_with_header(self):
415415
KORD,19990127, 23:00:00, 22:56:00, -0.5900, 1.7100, 4.6000, 0.0000, 280.0000"""
416416

417417
df = self.read_csv(StringIO(data), parse_dates={'nominal': [1, 2]})
418-
self.assert_(not isinstance(df.nominal[0], compat.string_types))
418+
self.assertNotIsInstance(df.nominal[0], compat.string_types)
419419

420420
ts_data = """\
421421
ID,date,nominalTime,actualTime,A,B,C,D,E
@@ -1006,8 +1006,7 @@ def test_read_csv_dataframe(self):
10061006
parse_dates=True)
10071007
self.assert_numpy_array_equal(df.columns, ['A', 'B', 'C', 'D'])
10081008
self.assertEqual(df.index.name, 'index')
1009-
self.assert_(isinstance(df.index[0], (datetime, np.datetime64,
1010-
Timestamp)))
1009+
self.assertIsInstance(df.index[0], (datetime, np.datetime64, Timestamp))
10111010
self.assertEqual(df.values.dtype, np.float64)
10121011
tm.assert_frame_equal(df, df2)
10131012

@@ -1016,8 +1015,7 @@ def test_read_csv_no_index_name(self):
10161015
df2 = self.read_table(self.csv2, sep=',', index_col=0,
10171016
parse_dates=True)
10181017
self.assert_numpy_array_equal(df.columns, ['A', 'B', 'C', 'D', 'E'])
1019-
self.assert_(isinstance(df.index[0], (datetime, np.datetime64,
1020-
Timestamp)))
1018+
self.assertIsInstance(df.index[0], (datetime, np.datetime64, Timestamp))
10211019
self.assertEqual(df.ix[:, ['A', 'B', 'C', 'D']].values.dtype, np.float64)
10221020
tm.assert_frame_equal(df, df2)
10231021

@@ -1441,13 +1439,13 @@ def test_multi_index_parse_dates(self):
14411439
20090103,three,c,4,5
14421440
"""
14431441
df = self.read_csv(StringIO(data), index_col=[0, 1], parse_dates=True)
1444-
self.assert_(isinstance(df.index.levels[0][0],
1445-
(datetime, np.datetime64, Timestamp)))
1442+
self.assertIsInstance(df.index.levels[0][0],
1443+
(datetime, np.datetime64, Timestamp))
14461444

14471445
# specify columns out of order!
14481446
df2 = self.read_csv(StringIO(data), index_col=[1, 0], parse_dates=True)
1449-
self.assert_(isinstance(df2.index.levels[1][0],
1450-
(datetime, np.datetime64, Timestamp)))
1447+
self.assertIsInstance(df2.index.levels[1][0],
1448+
(datetime, np.datetime64, Timestamp))
14511449

14521450
def test_skip_footer(self):
14531451
data = """A,B,C

‎pandas/io/tests/test_pytables.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3554,7 +3554,7 @@ def f():
35543554
# valid
35553555
result = store.select_column('df', 'index')
35563556
tm.assert_almost_equal(result.values, Series(df.index).values)
3557-
self.assert_(isinstance(result,Series))
3557+
self.assertIsInstance(result,Series)
35583558

35593559
# not a data indexable column
35603560
self.assertRaises(
@@ -3622,7 +3622,7 @@ def test_coordinates(self):
36223622
result = store.select('df', where=c)
36233623
expected = df.ix[3:4, :]
36243624
tm.assert_frame_equal(result, expected)
3625-
self.assert_(isinstance(c, Index))
3625+
self.assertIsInstance(c, Index)
36263626

36273627
# multiple tables
36283628
_maybe_remove(store, 'df1')

‎pandas/io/tests/test_stata.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -277,11 +277,11 @@ def test_encoding(self):
277277
if compat.PY3:
278278
expected = raw.kreis1849[0]
279279
self.assertEqual(result, expected)
280-
self.assert_(isinstance(result, compat.string_types))
280+
self.assertIsInstance(result, compat.string_types)
281281
else:
282282
expected = raw.kreis1849.str.decode("latin-1")[0]
283283
self.assertEqual(result, expected)
284-
self.assert_(isinstance(result, unicode))
284+
self.assertIsInstance(result, unicode)
285285

286286
def test_read_write_dta11(self):
287287
# skip_if_not_little_endian()

‎pandas/sparse/tests/test_sparse.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,7 @@ def _compare_with_dense(sp):
405405
def _compare(idx):
406406
dense_result = dense.take(idx).values
407407
sparse_result = sp.take(idx)
408-
self.assert_(isinstance(sparse_result, SparseSeries))
408+
self.assertIsInstance(sparse_result, SparseSeries)
409409
assert_almost_equal(dense_result, sparse_result.values.values)
410410

411411
_compare([1., 2., 3., 4., 5., 0.])
@@ -652,7 +652,7 @@ def test_dropna(self):
652652

653653
result = self.bseries.dropna()
654654
expected = self.bseries.to_dense().dropna()
655-
self.assert_(not isinstance(result, SparseSeries))
655+
self.assertNotIsInstance(result, SparseSeries)
656656
tm.assert_series_equal(result, expected)
657657

658658
def test_homogenize(self):

‎pandas/tests/test_compat.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,10 @@
1111
import nose
1212
import pandas.util.testing as tm
1313

14-
class TestBuiltinIterators(unittest.TestCase):
14+
class TestBuiltinIterators(tm.TestCase):
1515
def check_result(self, actual, expected, lengths):
1616
for (iter_res, list_res), exp, length in zip(actual, expected, lengths):
17-
self.assert_(not isinstance(iter_res, list))
17+
self.assertNotIsInstance(iter_res, list)
1818
tm.assert_isinstance(list_res, list)
1919
iter_res = list(iter_res)
2020
self.assertEqual(len(list_res), length)

‎pandas/tests/test_frame.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10427,11 +10427,11 @@ def _check_stat_op(self, name, alternative, frame=None, has_skipna=True,
1042710427
df = DataFrame({'b': date_range('1/1/2001', periods=2)})
1042810428
_f = getattr(df, name)
1042910429
result = _f()
10430-
self.assert_(isinstance(result, Series))
10430+
self.assertIsInstance(result, Series)
1043110431

1043210432
df['a'] = lrange(len(df))
1043310433
result = getattr(df, name)()
10434-
self.assert_(isinstance(result, Series))
10434+
self.assertIsInstance(result, Series)
1043510435
self.assert_(len(result))
1043610436

1043710437
if has_skipna:

‎pandas/tests/test_graphics.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -469,13 +469,13 @@ def test_xcompat(self):
469469
df = tm.makeTimeDataFrame()
470470
ax = df.plot(x_compat=True)
471471
lines = ax.get_lines()
472-
self.assert_(not isinstance(lines[0].get_xdata(), PeriodIndex))
472+
self.assertNotIsInstance(lines[0].get_xdata(), PeriodIndex)
473473

474474
tm.close()
475475
pd.plot_params['xaxis.compat'] = True
476476
ax = df.plot()
477477
lines = ax.get_lines()
478-
self.assert_(not isinstance(lines[0].get_xdata(), PeriodIndex))
478+
self.assertNotIsInstance(lines[0].get_xdata(), PeriodIndex)
479479

480480
tm.close()
481481
pd.plot_params['x_compat'] = False
@@ -488,7 +488,7 @@ def test_xcompat(self):
488488
with pd.plot_params.use('x_compat', True):
489489
ax = df.plot()
490490
lines = ax.get_lines()
491-
self.assert_(not isinstance(lines[0].get_xdata(), PeriodIndex))
491+
self.assertNotIsInstance(lines[0].get_xdata(), PeriodIndex)
492492

493493
tm.close()
494494
ax = df.plot()

‎pandas/tests/test_index.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,7 @@ def _check(op):
295295
index_result = op(index, element)
296296

297297
tm.assert_isinstance(index_result, np.ndarray)
298-
self.assert_(not isinstance(index_result, Index))
298+
self.assertNotIsInstance(index_result, Index)
299299
self.assert_numpy_array_equal(arr_result, index_result)
300300

301301
_check(operator.eq)
@@ -762,7 +762,7 @@ def test_boolean_cmp(self):
762762

763763
self.assert_(res.all())
764764
self.assertEqual(res.dtype, 'bool')
765-
self.assert_(not isinstance(res, Index))
765+
self.assertNotIsInstance(res, Index)
766766

767767
def test_get_level_values(self):
768768
result = self.strIndex.get_level_values(0)
@@ -808,35 +808,36 @@ def test_repr_roundtrip(self):
808808
tm.assert_index_equal(eval(repr(ind)), ind)
809809

810810
def check_is_index(self, i):
811-
self.assert_(isinstance(i, Index) and not isinstance(i, Float64Index))
811+
self.assertIsInstance(i, Index)
812+
self.assertNotIsInstance(i, Float64Index)
812813

813814
def check_coerce(self, a, b, is_float_index=True):
814815
self.assert_(a.equals(b))
815816
if is_float_index:
816-
self.assert_(isinstance(b, Float64Index))
817+
self.assertIsInstance(b, Float64Index)
817818
else:
818819
self.check_is_index(b)
819820

820821
def test_constructor(self):
821822

822823
# explicit construction
823824
index = Float64Index([1,2,3,4,5])
824-
self.assert_(isinstance(index, Float64Index))
825+
self.assertIsInstance(index, Float64Index)
825826
self.assert_((index.values == np.array([1,2,3,4,5],dtype='float64')).all())
826827
index = Float64Index(np.array([1,2,3,4,5]))
827-
self.assert_(isinstance(index, Float64Index))
828+
self.assertIsInstance(index, Float64Index)
828829
index = Float64Index([1.,2,3,4,5])
829-
self.assert_(isinstance(index, Float64Index))
830+
self.assertIsInstance(index, Float64Index)
830831
index = Float64Index(np.array([1.,2,3,4,5]))
831-
self.assert_(isinstance(index, Float64Index))
832+
self.assertIsInstance(index, Float64Index)
832833
self.assertEqual(index.dtype, object)
833834

834835
index = Float64Index(np.array([1.,2,3,4,5]),dtype=np.float32)
835-
self.assert_(isinstance(index, Float64Index))
836+
self.assertIsInstance(index, Float64Index)
836837
self.assertEqual(index.dtype, object)
837838

838839
index = Float64Index(np.array([1,2,3,4,5]),dtype=np.float32)
839-
self.assert_(isinstance(index, Float64Index))
840+
self.assertIsInstance(index, Float64Index)
840841
self.assertEqual(index.dtype, object)
841842

842843
# nan handling
@@ -1548,7 +1549,7 @@ def test_constructor_single_level(self):
15481549
labels=[[0, 1, 2, 3]],
15491550
names=['first'])
15501551
tm.assert_isinstance(single_level, Index)
1551-
self.assert_(not isinstance(single_level, MultiIndex))
1552+
self.assertNotIsInstance(single_level, MultiIndex)
15521553
self.assertEqual(single_level.name, 'first')
15531554

15541555
single_level = MultiIndex(levels=[['foo', 'bar', 'baz', 'qux']],

‎pandas/tests/test_internals.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,7 @@ def test_sparse(self):
378378
def test_sparse_mixed(self):
379379
mgr = create_blockmanager([get_sparse_ex1(),get_sparse_ex2(),get_float_ex()])
380380
self.assertEqual(len(mgr.blocks), 3)
381-
self.assert_(isinstance(mgr,BlockManager))
381+
self.assertIsInstance(mgr, BlockManager)
382382

383383
# what to test here?
384384

0 commit comments

Comments
 (0)