Skip to content

Commit 7087b64

Browse files
gfyounggfyoung
committed
CLN: Make test_categorical.py flake8-able
1 parent 5d5abf8 commit 7087b64

File tree

1 file changed

+28
-29
lines changed

1 file changed

+28
-29
lines changed

‎pandas/tests/indexes/test_category.py

+28-29
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,5 @@
11
# -*- coding: utf-8 -*-
22

3-
# TODO(wesm): fix long line flake8 issues
4-
# flake8: noqa
5-
63
import pandas.util.testing as tm
74
from pandas.indexes.api import Index, CategoricalIndex
85
from .common import Base
@@ -215,7 +212,8 @@ def test_map(self):
215212

216213
# GH 12766: Return an index not an array
217214
tm.assert_index_equal(ci.map(lambda x: 1),
218-
Index(np.array([1] * 5, dtype=np.int64), name='XXX'))
215+
Index(np.array([1] * 5, dtype=np.int64),
216+
name='XXX'))
219217

220218
# change categories dtype
221219
ci = pd.CategoricalIndex(list('ABABC'), categories=list('BAC'),
@@ -225,7 +223,8 @@ def f(x):
225223
return {'A': 10, 'B': 20, 'C': 30}.get(x)
226224

227225
result = ci.map(f)
228-
exp = pd.CategoricalIndex([10, 20, 10, 20, 30], categories=[20, 10, 30],
226+
exp = pd.CategoricalIndex([10, 20, 10, 20, 30],
227+
categories=[20, 10, 30],
229228
ordered=False)
230229
tm.assert_index_equal(result, exp)
231230

@@ -589,10 +588,10 @@ def test_string_categorical_index_repr(self):
589588
# short
590589
idx = pd.CategoricalIndex(['a', 'bb', 'ccc'])
591590
if PY3:
592-
expected = u"""CategoricalIndex(['a', 'bb', 'ccc'], categories=['a', 'bb', 'ccc'], ordered=False, dtype='category')"""
591+
expected = u"""CategoricalIndex(['a', 'bb', 'ccc'], categories=['a', 'bb', 'ccc'], ordered=False, dtype='category')""" # noqa
593592
self.assertEqual(repr(idx), expected)
594593
else:
595-
expected = u"""CategoricalIndex([u'a', u'bb', u'ccc'], categories=[u'a', u'bb', u'ccc'], ordered=False, dtype='category')"""
594+
expected = u"""CategoricalIndex([u'a', u'bb', u'ccc'], categories=[u'a', u'bb', u'ccc'], ordered=False, dtype='category')""" # noqa
596595
self.assertEqual(unicode(idx), expected)
597596

598597
# multiple lines
@@ -601,15 +600,15 @@ def test_string_categorical_index_repr(self):
601600
expected = u"""CategoricalIndex(['a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a',
602601
'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb',
603602
'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc'],
604-
categories=['a', 'bb', 'ccc'], ordered=False, dtype='category')"""
603+
categories=['a', 'bb', 'ccc'], ordered=False, dtype='category')""" # noqa
605604

606605
self.assertEqual(repr(idx), expected)
607606
else:
608607
expected = u"""CategoricalIndex([u'a', u'bb', u'ccc', u'a', u'bb', u'ccc', u'a', u'bb',
609608
u'ccc', u'a', u'bb', u'ccc', u'a', u'bb', u'ccc', u'a',
610609
u'bb', u'ccc', u'a', u'bb', u'ccc', u'a', u'bb', u'ccc',
611610
u'a', u'bb', u'ccc', u'a', u'bb', u'ccc'],
612-
categories=[u'a', u'bb', u'ccc'], ordered=False, dtype='category')"""
611+
categories=[u'a', u'bb', u'ccc'], ordered=False, dtype='category')""" # noqa
613612

614613
self.assertEqual(unicode(idx), expected)
615614

@@ -619,7 +618,7 @@ def test_string_categorical_index_repr(self):
619618
expected = u"""CategoricalIndex(['a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a',
620619
...
621620
'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc'],
622-
categories=['a', 'bb', 'ccc'], ordered=False, dtype='category', length=300)"""
621+
categories=['a', 'bb', 'ccc'], ordered=False, dtype='category', length=300)""" # noqa
623622

624623
self.assertEqual(repr(idx), expected)
625624
else:
@@ -628,7 +627,7 @@ def test_string_categorical_index_repr(self):
628627
...
629628
u'ccc', u'a', u'bb', u'ccc', u'a', u'bb', u'ccc', u'a',
630629
u'bb', u'ccc'],
631-
categories=[u'a', u'bb', u'ccc'], ordered=False, dtype='category', length=300)"""
630+
categories=[u'a', u'bb', u'ccc'], ordered=False, dtype='category', length=300)""" # noqa
632631

633632
self.assertEqual(unicode(idx), expected)
634633

@@ -637,23 +636,23 @@ def test_string_categorical_index_repr(self):
637636
if PY3:
638637
expected = u"""CategoricalIndex(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l',
639638
'm', 'm', 'o'],
640-
categories=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', ...], ordered=False, dtype='category')"""
639+
categories=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', ...], ordered=False, dtype='category')""" # noqa
641640

642641
self.assertEqual(repr(idx), expected)
643642
else:
644643
expected = u"""CategoricalIndex([u'a', u'b', u'c', u'd', u'e', u'f', u'g', u'h', u'i', u'j',
645644
u'k', u'l', u'm', u'm', u'o'],
646-
categories=[u'a', u'b', u'c', u'd', u'e', u'f', u'g', u'h', ...], ordered=False, dtype='category')"""
645+
categories=[u'a', u'b', u'c', u'd', u'e', u'f', u'g', u'h', ...], ordered=False, dtype='category')""" # noqa
647646

648647
self.assertEqual(unicode(idx), expected)
649648

650649
# short
651650
idx = pd.CategoricalIndex([u'あ', u'いい', u'ううう'])
652651
if PY3:
653-
expected = u"""CategoricalIndex(['あ', 'いい', 'ううう'], categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')"""
652+
expected = u"""CategoricalIndex(['あ', 'いい', 'ううう'], categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')""" # noqa
654653
self.assertEqual(repr(idx), expected)
655654
else:
656-
expected = u"""CategoricalIndex([u'あ', u'いい', u'ううう'], categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category')"""
655+
expected = u"""CategoricalIndex([u'あ', u'いい', u'ううう'], categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category')""" # noqa
657656
self.assertEqual(unicode(idx), expected)
658657

659658
# multiple lines
@@ -662,15 +661,15 @@ def test_string_categorical_index_repr(self):
662661
expected = u"""CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ',
663662
'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
664663
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
665-
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')"""
664+
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')""" # noqa
666665

667666
self.assertEqual(repr(idx), expected)
668667
else:
669668
expected = u"""CategoricalIndex([u'あ', u'いい', u'ううう', u'あ', u'いい', u'ううう', u'あ', u'いい',
670669
u'ううう', u'あ', u'いい', u'ううう', u'あ', u'いい', u'ううう', u'あ',
671670
u'いい', u'ううう', u'あ', u'いい', u'ううう', u'あ', u'いい', u'ううう',
672671
u'あ', u'いい', u'ううう', u'あ', u'いい', u'ううう'],
673-
categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category')"""
672+
categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category')""" # noqa
674673

675674
self.assertEqual(unicode(idx), expected)
676675

@@ -680,7 +679,7 @@ def test_string_categorical_index_repr(self):
680679
expected = u"""CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ',
681680
...
682681
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
683-
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)"""
682+
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)""" # noqa
684683

685684
self.assertEqual(repr(idx), expected)
686685
else:
@@ -689,7 +688,7 @@ def test_string_categorical_index_repr(self):
689688
...
690689
u'ううう', u'あ', u'いい', u'ううう', u'あ', u'いい', u'ううう', u'あ',
691690
u'いい', u'ううう'],
692-
categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category', length=300)"""
691+
categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category', length=300)""" # noqa
693692

694693
self.assertEqual(unicode(idx), expected)
695694

@@ -698,13 +697,13 @@ def test_string_categorical_index_repr(self):
698697
if PY3:
699698
expected = u"""CategoricalIndex(['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', 'け', 'こ', 'さ', 'し',
700699
'す', 'せ', 'そ'],
701-
categories=['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', ...], ordered=False, dtype='category')"""
700+
categories=['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', ...], ordered=False, dtype='category')""" # noqa
702701

703702
self.assertEqual(repr(idx), expected)
704703
else:
705704
expected = u"""CategoricalIndex([u'あ', u'い', u'う', u'え', u'お', u'か', u'き', u'く', u'け', u'こ',
706705
u'さ', u'し', u'す', u'せ', u'そ'],
707-
categories=[u'あ', u'い', u'う', u'え', u'お', u'か', u'き', u'く', ...], ordered=False, dtype='category')"""
706+
categories=[u'あ', u'い', u'う', u'え', u'お', u'か', u'き', u'く', ...], ordered=False, dtype='category')""" # noqa
708707

709708
self.assertEqual(unicode(idx), expected)
710709

@@ -714,10 +713,10 @@ def test_string_categorical_index_repr(self):
714713
# short
715714
idx = pd.CategoricalIndex([u'あ', u'いい', u'ううう'])
716715
if PY3:
717-
expected = u"""CategoricalIndex(['あ', 'いい', 'ううう'], categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')"""
716+
expected = u"""CategoricalIndex(['あ', 'いい', 'ううう'], categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')""" # noqa
718717
self.assertEqual(repr(idx), expected)
719718
else:
720-
expected = u"""CategoricalIndex([u'あ', u'いい', u'ううう'], categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category')"""
719+
expected = u"""CategoricalIndex([u'あ', u'いい', u'ううう'], categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category')""" # noqa
721720
self.assertEqual(unicode(idx), expected)
722721

723722
# multiple lines
@@ -727,7 +726,7 @@ def test_string_categorical_index_repr(self):
727726
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
728727
'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
729728
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
730-
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')"""
729+
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')""" # noqa
731730

732731
self.assertEqual(repr(idx), expected)
733732
else:
@@ -736,7 +735,7 @@ def test_string_categorical_index_repr(self):
736735
u'いい', u'ううう', u'あ', u'いい', u'ううう', u'あ',
737736
u'いい', u'ううう', u'あ', u'いい', u'ううう', u'あ',
738737
u'いい', u'ううう', u'あ', u'いい', u'ううう'],
739-
categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category')"""
738+
categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category')""" # noqa
740739

741740
self.assertEqual(unicode(idx), expected)
742741

@@ -748,7 +747,7 @@ def test_string_categorical_index_repr(self):
748747
...
749748
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
750749
'あ', 'いい', 'ううう'],
751-
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)"""
750+
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)""" # noqa
752751

753752
self.assertEqual(repr(idx), expected)
754753
else:
@@ -757,7 +756,7 @@ def test_string_categorical_index_repr(self):
757756
...
758757
u'ううう', u'あ', u'いい', u'ううう', u'あ', u'いい',
759758
u'ううう', u'あ', u'いい', u'ううう'],
760-
categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category', length=300)"""
759+
categories=[u'あ', u'いい', u'ううう'], ordered=False, dtype='category', length=300)""" # noqa
761760

762761
self.assertEqual(unicode(idx), expected)
763762

@@ -766,13 +765,13 @@ def test_string_categorical_index_repr(self):
766765
if PY3:
767766
expected = u"""CategoricalIndex(['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', 'け', 'こ',
768767
'さ', 'し', 'す', 'せ', 'そ'],
769-
categories=['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', ...], ordered=False, dtype='category')"""
768+
categories=['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', ...], ordered=False, dtype='category')""" # noqa
770769

771770
self.assertEqual(repr(idx), expected)
772771
else:
773772
expected = u"""CategoricalIndex([u'あ', u'い', u'う', u'え', u'お', u'か', u'き', u'く',
774773
u'け', u'こ', u'さ', u'し', u'す', u'せ', u'そ'],
775-
categories=[u'あ', u'い', u'う', u'え', u'お', u'か', u'き', u'く', ...], ordered=False, dtype='category')"""
774+
categories=[u'あ', u'い', u'う', u'え', u'お', u'か', u'き', u'く', ...], ordered=False, dtype='category')""" # noqa
776775

777776
self.assertEqual(unicode(idx), expected)
778777

0 commit comments

Comments
 (0)