MCPcopy Index your code
hub / github.com/quantopian/zipline / test_setitem_scalar

Method test_setitem_scalar

tests/test_labelarray.py:341–379  ·  view source on GitHub ↗
(self, val, missing_value)

Source from the content-addressed store, hash-verified

339 missing_value=['', 'a', 'not in the array', None],
340 )
341 def test_setitem_scalar(self, val, missing_value):
342 arr = LabelArray(self.strs, missing_value=missing_value)
343
344 if not arr.has_label(val):
345 self.assertTrue(
346 (val == 'not in the array')
347 or (val is None and missing_value is not None)
348 )
349 for slicer in [(0, 0), (0, 1), 1]:
350 with self.assertRaises(ValueError):
351 arr[slicer] = val
352 return
353
354 arr[0, 0] = val
355 self.assertEqual(arr[0, 0], val)
356
357 arr[0, 1] = val
358 self.assertEqual(arr[0, 1], val)
359
360 arr[1] = val
361 if val == missing_value:
362 self.assertTrue(arr.is_missing()[1].all())
363 else:
364 self.assertTrue((arr[1] == val).all())
365 self.assertTrue((arr[1].as_string_array() == val).all())
366
367 arr[:, -1] = val
368 if val == missing_value:
369 self.assertTrue(arr.is_missing()[:, -1].all())
370 else:
371 self.assertTrue((arr[:, -1] == val).all())
372 self.assertTrue((arr[:, -1].as_string_array() == val).all())
373
374 arr[:] = val
375 if val == missing_value:
376 self.assertTrue(arr.is_missing().all())
377 else:
378 self.assertFalse(arr.is_missing().any())
379 self.assertTrue((arr == val).all())
380
381 def test_setitem_array(self):
382 arr = LabelArray(self.strs, missing_value=None)

Callers

nothing calls this directly

Calls 4

has_labelMethod · 0.95
is_missingMethod · 0.95
LabelArrayClass · 0.90
as_string_arrayMethod · 0.80

Tested by

no test coverage detected