def test_frame_via_fill_value_loc_e1(self) -> None:
f1 = Frame(np.arange(12).reshape(4, 3),
index=tuple('abcd'),
columns=tuple('xyz'))
s1 = f1.via_fill_value(-1)['y']
self.assertEqual(s1.to_pairs(),
(('a', 1), ('b', 4), ('c', 7), ('d', 10)))
f2 = f1.via_fill_value(-1)[['y', 'w']]
self.assertEqual(f2.to_pairs(),
(('y', (('a', 1), ('b', 4), ('c', 7), ('d', 10))),
('w', (('a', -1), ('b', -1), ('c', -1), ('d', -1)))))
def test_frame_via_fill_value_loc_h(self) -> None:
f1 = Frame(np.arange(12).reshape(4, 3),
index=tuple('abcd'),
columns=tuple('xyz'))
f2 = f1.via_fill_value(-1).loc[['d', 'e'], 'w']
self.assertEqual(f2.to_pairs(), (('d', -1), ('e', -1)))
def test_frame_via_fill_value_loc_d(self) -> None:
f1 = Frame(np.arange(12).reshape(4, 3),
index=tuple('abcd'),
columns=tuple('xyz'))
s1 = f1.via_fill_value(-1).loc['q', ['w', 'y', 'z']]
self.assertEqual(s1.to_pairs(), (('w', -1), ('y', -1), ('z', -1)))
self.assertEqual(s1.name, 'q')
def test_frame_via_fill_value_loc_a(self) -> None:
f1 = Frame(np.arange(12).reshape(4, 3),
index=tuple('abcd'),
columns=tuple('xyz'),
name='foo')
f2 = f1.via_fill_value(-1).loc['c':, ['w', 'x']]
self.assertEqual(f2.to_pairs(), (('w', (('c', -1), ('d', -1))),
('x', (('c', 6), ('d', 9)))))
def test_frame_via_fill_value_loc_b(self) -> None:
f1 = Frame(np.arange(12).reshape(4, 3),
index=tuple('abcd'),
columns=tuple('xyz'),
name='foo')
self.assertEqual(f1.via_fill_value(-1).loc['a', 'z'], 2)
self.assertEqual(f1.via_fill_value(-1).loc['a', 'w'], -1)
self.assertEqual(f1.name, 'foo')
def test_frame_via_fill_value_e(self) -> None:
f1 = FrameGO(index=range(5))
f1.via_fill_value(0)['a'] = Series([10, 20], index=(2, 4))
f1.via_fill_value(-1)['b'] = Series([10, 20], index=(0, 1))
self.assertEqual(f1.to_pairs(),
(('a', ((0, 0), (1, 0), (2, 10), (3, 0), (4, 20))),
('b', ((0, 10), (1, 20), (2, -1), (3, -1),
(4, -1)))))
f2 = Frame(index=range(5))
with self.assertRaises(TypeError):
f2.via_fill_value(0)['a'] = range(5) # type: ignore #pylint: disable=E1137