def test_parse_kwargs_list(): x = [np.random.random((3, 3))] * 2 kwargs = {'label': ['Group A', 'Group B']} assert helpers.parse_kwargs(x, kwargs) == [{ 'label': 'Group A' }, { 'label': 'Group B' }]
def test_parse_kwargs_array(): x = [np.random.random((3, 3))] kwargs = {'label': ['Group A']} assert helpers.parse_kwargs(x, kwargs) == [{'label': 'Group A'}]