|
4 | 4 | import nose
|
5 | 5 |
|
6 | 6 | import numpy as np
|
| 7 | +from numpy import (iinfo, int8, int16, int32, int64, |
| 8 | + uint8, uint16, uint32,) |
7 | 9 |
|
8 | 10 | import pandas as pd
|
9 | 11 | from pandas import date_range, Index
|
@@ -402,26 +404,28 @@ def test_downcast(self):
|
402 | 404 | tm.assert_numpy_array_equal(res, expected)
|
403 | 405 |
|
404 | 406 | def test_downcast_limits(self):
|
405 |
| - # Test the limits of each downcast. Bug: #14401. |
| 407 | + # Test the limits of each downcast. Bug: #14401. |
| 408 | + i = 'integer' |
| 409 | + u = 'unsigned' |
406 | 410 | dtype_downcast_min_max = [
|
407 |
| - ('int8', 'integer', [np.iinfo(np.int8).min, np.iinfo(np.int8).max]), |
408 |
| - ('int16', 'integer', [np.iinfo(np.int16).min, np.iinfo(np.int16).max]), |
409 |
| - ('int32', 'integer', [np.iinfo(np.int32).min, np.iinfo(np.int32).max]), |
410 |
| - ('int64', 'integer', [np.iinfo(np.int64).min, np.iinfo(np.int64).max]), |
411 |
| - ('uint8', 'unsigned', [np.iinfo(np.uint8).min, np.iinfo(np.uint8).max]), |
412 |
| - ('uint16', 'unsigned', [np.iinfo(np.uint16).min, np.iinfo(np.uint16).max]), |
413 |
| - ('uint32', 'unsigned', [np.iinfo(np.uint32).min, np.iinfo(np.uint32).max]), |
414 |
| - # ('uint64', 'unsigned', [np.iinfo(np.uint64).min, np.iinfo(np.uint64).max]), |
415 |
| - ('int16', 'integer', [np.iinfo(np.int8).min, np.iinfo(np.int8).max + 1]), |
416 |
| - ('int32', 'integer', [np.iinfo(np.int16).min, np.iinfo(np.int16).max + 1]), |
417 |
| - ('int64', 'integer', [np.iinfo(np.int32).min, np.iinfo(np.int32).max + 1]), |
418 |
| - ('int16', 'integer', [np.iinfo(np.int8).min - 1, np.iinfo(np.int16).max]), |
419 |
| - ('int32', 'integer', [np.iinfo(np.int16).min - 1, np.iinfo(np.int32).max]), |
420 |
| - ('int64', 'integer', [np.iinfo(np.int32).min - 1, np.iinfo(np.int64).max]), |
421 |
| - ('uint16', 'unsigned', [np.iinfo(np.uint8).min, np.iinfo(np.uint8).max + 1]), |
422 |
| - ('uint32', 'unsigned', [np.iinfo(np.uint16).min, np.iinfo(np.uint16).max + 1]), |
423 |
| - # ('uint64', 'unsigned', [np.iinfo(np.uint32).min, np.iinfo(np.uint32).max + 1]), |
424 |
| - ] |
| 411 | + ('int8', i, [iinfo(int8).min, iinfo(int8).max]), |
| 412 | + ('int16', i, [iinfo(int16).min, iinfo(int16).max]), |
| 413 | + ('int32', i, [iinfo(int32).min, iinfo(int32).max]), |
| 414 | + ('int64', i, [iinfo(int64).min, iinfo(int64).max]), |
| 415 | + ('uint8', u, [iinfo(uint8).min, iinfo(uint8).max]), |
| 416 | + ('uint16', u, [iinfo(uint16).min, iinfo(uint16).max]), |
| 417 | + ('uint32', u, [iinfo(uint32).min, iinfo(uint32).max]), |
| 418 | + # ('uint64', u, [iinfo(uint64).min, iinfo(uint64).max]), |
| 419 | + ('int16', i, [iinfo(int8).min, iinfo(int8).max + 1]), |
| 420 | + ('int32', i, [iinfo(int16).min, iinfo(int16).max + 1]), |
| 421 | + ('int64', i, [iinfo(int32).min, iinfo(int32).max + 1]), |
| 422 | + ('int16', i, [iinfo(int8).min - 1, iinfo(int16).max]), |
| 423 | + ('int32', i, [iinfo(int16).min - 1, iinfo(int32).max]), |
| 424 | + ('int64', i, [iinfo(int32).min - 1, iinfo(int64).max]), |
| 425 | + ('uint16', u, [iinfo(uint8).min, iinfo(uint8).max + 1]), |
| 426 | + ('uint32', u, [iinfo(uint16).min, iinfo(uint16).max + 1]), |
| 427 | + # ('uint64', u, [iinfo(uint32).min, iinfo(uint32).max + 1]), |
| 428 | + ] |
425 | 429 |
|
426 | 430 | for dtype, downcast, min_max in dtype_downcast_min_max:
|
427 | 431 | series = pd.to_numeric(pd.Series(min_max), downcast=downcast)
|
|
0 commit comments