wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-03 08:34:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

TST: add tests for inferred width labelarray

Browse Source
This commit is contained in:
Joe Jevnik
2017-02-07 16:28:13 -05:00
parent d378c9fca3
commit 735d98fb6e
1 changed files with 36 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_labelarray.py
+36 -5
View File
@@ -351,24 +351,55 @@ class LabelArrayTestCase(ZiplineTestCase):
)
]
# uint8
# uint8
categories = create_categories(8, plus_one=False)
arr = LabelArray([], missing_value='', categories=categories)
arr = LabelArray(
[],
missing_value=categories[0],
categories=categories,
)
self.assertEqual(arr.itemsize, 1)
# uint8 inference
arr = LabelArray(categories, missing_value=categories[0])
self.assertEqual(arr.itemsize, 1)
# just over uint8
categories = create_categories(8, plus_one=True)
arr = LabelArray([], missing_value='', categories=categories)
arr = LabelArray(
[],
missing_value=categories[0],
categories=categories,
)
self.assertEqual(arr.itemsize, 2)
# uint16 inference
arr = LabelArray(categories, missing_value=categories[0])
self.assertEqual(arr.itemsize, 2)
# fits in uint16
categories = create_categories(16, plus_one=False)
arr = LabelArray([], missing_value='', categories=categories)
arr = LabelArray(
[], missing_value=categories[0],
categories=categories,
)
self.assertEqual(arr.itemsize, 2)
# uint16 inference
arr = LabelArray(categories, missing_value=categories[0])
self.assertEqual(arr.itemsize, 2)
# just over uint16
categories = create_categories(16, plus_one=True)
arr = LabelArray([], missing_value='', categories=categories)
arr = LabelArray(
[],
missing_value=categories[0],
categories=categories,
)
self.assertEqual(arr.itemsize, 4)
# uint32 inference
arr = LabelArray(categories, missing_value=categories[0])
self.assertEqual(arr.itemsize, 4)
# NOTE: we could do this for 32 and 64; however, no one has enough RAM