@@ -184,10 +184,6 @@ def wrapper(x):
184
184
185
185
class SafeForSparse (object ):
186
186
187
- @classmethod
188
- def assert_panel_equal (cls , x , y ):
189
- assert_panel_equal (x , y )
190
-
191
187
def test_get_axis (self ):
192
188
assert (self .panel ._get_axis (0 ) is self .panel .items )
193
189
assert (self .panel ._get_axis (1 ) is self .panel .major_axis )
@@ -353,13 +349,11 @@ def check_op(op, name):
353
349
raise
354
350
355
351
def test_combinePanel (self ):
356
- with catch_warnings (record = True ):
357
- result = self .panel .add (self .panel )
358
- assert_panel_equal (result , self .panel * 2 )
352
+ with catch_warnings (record = True ):result = self .panel .add (self .panel )
353
+ assert_panel_equal (result , self .panel * 2 )
359
354
360
355
def test_neg (self ):
361
- with catch_warnings (record = True ):
362
- assert_panel_equal (- self .panel , self .panel * - 1 )
356
+ assert_panel_equal (- self .panel , self .panel * - 1 )
363
357
364
358
# issue 7692
365
359
def test_raise_when_not_implemented (self ):
@@ -379,26 +373,25 @@ def test_select(self):
379
373
with catch_warnings (record = True ):
380
374
p = self .panel
381
375
382
- # select items
383
- result = p .select (lambda x : x in ('ItemA' , 'ItemC' ), axis = 'items' )
384
- expected = p .reindex (items = ['ItemA' , 'ItemC' ])
385
- assert_panel_equal (result , expected )
376
+ # select items
377
+ result = p .select (lambda x : x in ('ItemA' , 'ItemC' ), axis = 'items' )
378
+ expected = p .reindex (items = ['ItemA' , 'ItemC' ])
379
+ assert_panel_equal (result , expected )
386
380
387
- # select major_axis
388
- result = p .select (lambda x : x >= datetime (
389
- 2000 , 1 , 15 ), axis = 'major' )
390
- new_major = p .major_axis [p .major_axis >= datetime (2000 , 1 , 15 )]
391
- expected = p .reindex (major = new_major )
392
- assert_panel_equal (result , expected )
381
+ # select major_axis
382
+ result = p .select (lambda x : x >= datetime (2000 , 1 , 15 ), axis = 'major' )
383
+ new_major = p .major_axis [p .major_axis >= datetime (2000 , 1 , 15 )]
384
+ expected = p .reindex (major = new_major )
385
+ assert_panel_equal (result , expected )
393
386
394
- # select minor_axis
395
- result = p .select (lambda x : x in ('D' , 'A' ), axis = 2 )
396
- expected = p .reindex (minor = ['A' , 'D' ])
397
- assert_panel_equal (result , expected )
387
+ # select minor_axis
388
+ result = p .select (lambda x : x in ('D' , 'A' ), axis = 2 )
389
+ expected = p .reindex (minor = ['A' , 'D' ])
390
+ assert_panel_equal (result , expected )
398
391
399
- # corner case, empty thing
400
- result = p .select (lambda x : x in ('foo' , ), axis = 'items' )
401
- assert_panel_equal (result , p .reindex (items = []))
392
+ # corner case, empty thing
393
+ result = p .select (lambda x : x in ('foo' , ), axis = 'items' )
394
+ assert_panel_equal (result , p .reindex (items = []))
402
395
403
396
def test_get_value (self ):
404
397
for item in self .panel .items :
@@ -906,10 +899,6 @@ def test_set_value(self):
906
899
class TestPanel (PanelTests , CheckIndexing , SafeForLongAndSparse ,
907
900
SafeForSparse ):
908
901
909
- @classmethod
910
- def assert_panel_equal (cls , x , y ):
911
- assert_panel_equal (x , y )
912
-
913
902
def setup_method (self , method ):
914
903
self .panel = make_test_panel ()
915
904
self .panel .major_axis .name = None
@@ -2155,8 +2144,8 @@ def test_multiindex_get(self):
2155
2144
assert (f1 .items == [1 , 2 ]).all ()
2156
2145
assert (f2 .items == [1 , 2 ]).all ()
2157
2146
2158
- ind = MultiIndex .from_tuples ([('a' , 1 ), ('a' , 2 ), ('b' , 1 )],
2159
- names = ['first' , 'second' ])
2147
+ MultiIndex .from_tuples ([('a' , 1 ), ('a' , 2 ), ('b' , 1 )],
2148
+ names = ['first' , 'second' ])
2160
2149
2161
2150
def test_multiindex_blocks (self ):
2162
2151
with catch_warnings (record = True ):
@@ -2461,13 +2450,15 @@ def test_sort_values(self):
2461
2450
pytest .raises (NotImplementedError , self .panel .sort_values , 'ItemA' )
2462
2451
2463
2452
2464
- class TestLongPanel (object ):
2453
+ class TestPanelFrame (object ):
2465
2454
"""
2466
- LongPanel no longer exists, but.. .
2455
+ Check that conversions to and from Panel to DataFrame work .
2467
2456
"""
2468
2457
2469
- def setup_method (self , method ):
2470
- panel = make_test_panel ()
2458
+ def setUp (self ):
2459
+ panel = tm .makePanel ()
2460
+ tm .add_nans (panel )
2461
+
2471
2462
self .panel = panel .to_frame ()
2472
2463
self .unfiltered_panel = panel .to_frame (filter_observations = False )
2473
2464
0 commit comments