def test_tolist(self): array = MappedArray([0] * 5 + [1] * 5, values_mapping={ 0: '-', 1: 'Warning' }) self.assertEqual(array.tolist(), ['-'] * 5 + ['Warning'] * 5) array[2] = np.ma.masked self.assertEqual(array.tolist(), ['-'] * 2 + [None] + ['-'] * 2 + ['Warning'] * 5)
def test_tolist(self): array = MappedArray([0] * 5 + [1] * 5, values_mapping={0: '-', 1: 'Warning'}) self.assertEqual(array.tolist(), ['-'] * 5 + ['Warning'] * 5) array[2] = np.ma.masked self.assertEqual(array.tolist(), ['-'] * 2 + [None] + ['-'] * 2 + ['Warning'] * 5)