@@ -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 :
@@ -412,12 +405,11 @@ def test_get_value(self):
412
405
413
406
def test_abs (self ):
414
407
415
- with catch_warnings (record = True ):
416
- result = self .panel .abs ()
417
- result2 = abs (self .panel )
418
- expected = np .abs (self .panel )
419
- assert_panel_equal (result , expected )
420
- assert_panel_equal (result2 , expected )
408
+ with catch_warnings (record = True ):result = self .panel .abs ()
409
+ result2 = abs (self .panel )
410
+ expected = np .abs (self .panel )
411
+ assert_panel_equal (result , expected )
412
+ assert_panel_equal (result2 , expected )
421
413
422
414
df = self .panel ['ItemA' ]
423
415
result = df .abs ()
@@ -906,10 +898,6 @@ def test_set_value(self):
906
898
class TestPanel (PanelTests , CheckIndexing , SafeForLongAndSparse ,
907
899
SafeForSparse ):
908
900
909
- @classmethod
910
- def assert_panel_equal (cls , x , y ):
911
- assert_panel_equal (x , y )
912
-
913
901
def setup_method (self , method ):
914
902
self .panel = make_test_panel ()
915
903
self .panel .major_axis .name = None
@@ -2103,38 +2091,36 @@ def test_pct_change(self):
2103
2091
assert_panel_equal (result , expected )
2104
2092
2105
2093
def test_round (self ):
2106
- with catch_warnings (record = True ):
2107
- values = [[[- 3.2 , 2.2 ], [0 , - 4.8213 ], [3.123 , 123.12 ],
2108
- [- 1566.213 , 88.88 ], [- 12 , 94.5 ]],
2109
- [[- 5.82 , 3.5 ], [6.21 , - 73.272 ], [- 9.087 , 23.12 ],
2110
- [272.212 , - 99.99 ], [23 , - 76.5 ]]]
2111
- evalues = [[[float (np .around (i )) for i in j ] for j in k ]
2112
- for k in values ]
2113
- p = Panel (values , items = ['Item1' , 'Item2' ],
2114
- major_axis = date_range ('1/1/2000' , periods = 5 ),
2115
- minor_axis = ['A' , 'B' ])
2116
- expected = Panel (evalues , items = ['Item1' , 'Item2' ],
2117
- major_axis = date_range ('1/1/2000' , periods = 5 ),
2118
- minor_axis = ['A' , 'B' ])
2119
- result = p .round ()
2120
- assert_panel_equal (expected , result )
2094
+ with catch_warnings (record = True ):values = [[[- 3.2 , 2.2 ], [0 , - 4.8213 ], [3.123 , 123.12 ],
2095
+ [- 1566.213 , 88.88 ], [- 12 , 94.5 ]],
2096
+ [[- 5.82 , 3.5 ], [6.21 , - 73.272 ], [- 9.087 , 23.12 ],
2097
+ [272.212 , - 99.99 ], [23 , - 76.5 ]]]
2098
+ evalues = [[[float (np .around (i )) for i in j ] for j in k ]
2099
+ for k in values ]
2100
+ p = Panel (values , items = ['Item1' , 'Item2' ],
2101
+ major_axis = date_range ('1/1/2000' , periods = 5 ),
2102
+ minor_axis = ['A' , 'B' ])
2103
+ expected = Panel (evalues , items = ['Item1' , 'Item2' ],
2104
+ major_axis = date_range ('1/1/2000' , periods = 5 ),
2105
+ minor_axis = ['A' , 'B' ])
2106
+ result = p .round ()
2107
+ assert_panel_equal (expected , result )
2121
2108
2122
2109
def test_numpy_round (self ):
2123
- with catch_warnings (record = True ):
2124
- values = [[[- 3.2 , 2.2 ], [0 , - 4.8213 ], [3.123 , 123.12 ],
2125
- [- 1566.213 , 88.88 ], [- 12 , 94.5 ]],
2126
- [[- 5.82 , 3.5 ], [6.21 , - 73.272 ], [- 9.087 , 23.12 ],
2127
- [272.212 , - 99.99 ], [23 , - 76.5 ]]]
2128
- evalues = [[[float (np .around (i )) for i in j ] for j in k ]
2129
- for k in values ]
2130
- p = Panel (values , items = ['Item1' , 'Item2' ],
2131
- major_axis = date_range ('1/1/2000' , periods = 5 ),
2132
- minor_axis = ['A' , 'B' ])
2133
- expected = Panel (evalues , items = ['Item1' , 'Item2' ],
2134
- major_axis = date_range ('1/1/2000' , periods = 5 ),
2135
- minor_axis = ['A' , 'B' ])
2136
- result = np .round (p )
2137
- assert_panel_equal (expected , result )
2110
+ with catch_warnings (record = True ):values = [[[- 3.2 , 2.2 ], [0 , - 4.8213 ], [3.123 , 123.12 ],
2111
+ [- 1566.213 , 88.88 ], [- 12 , 94.5 ]],
2112
+ [[- 5.82 , 3.5 ], [6.21 , - 73.272 ], [- 9.087 , 23.12 ],
2113
+ [272.212 , - 99.99 ], [23 , - 76.5 ]]]
2114
+ evalues = [[[float (np .around (i )) for i in j ] for j in k ]
2115
+ for k in values ]
2116
+ p = Panel (values , items = ['Item1' , 'Item2' ],
2117
+ major_axis = date_range ('1/1/2000' , periods = 5 ),
2118
+ minor_axis = ['A' , 'B' ])
2119
+ expected = Panel (evalues , items = ['Item1' , 'Item2' ],
2120
+ major_axis = date_range ('1/1/2000' , periods = 5 ),
2121
+ minor_axis = ['A' , 'B' ])
2122
+ result = np .round (p )
2123
+ assert_panel_equal (expected , result )
2138
2124
2139
2125
msg = "the 'out' parameter is not supported"
2140
2126
tm .assert_raises_regex (ValueError , msg , np .round , p , out = p )
@@ -2461,13 +2447,15 @@ def test_sort_values(self):
2461
2447
pytest .raises (NotImplementedError , self .panel .sort_values , 'ItemA' )
2462
2448
2463
2449
2464
- class TestLongPanel (object ):
2450
+ class TestPanelFrame (object ):
2465
2451
"""
2466
- LongPanel no longer exists, but.. .
2452
+ Check that conversions to and from Panel to DataFrame work .
2467
2453
"""
2468
2454
2469
- def setup_method (self , method ):
2470
- panel = make_test_panel ()
2455
+ def setUp (self ):
2456
+ panel = tm .makePanel ()
2457
+ tm .add_nans (panel )
2458
+
2471
2459
self .panel = panel .to_frame ()
2472
2460
self .unfiltered_panel = panel .to_frame (filter_observations = False )
2473
2461
0 commit comments