def test_apply_with_constant_string():
features = pd.DataFrame(index=range(3))
text_manager = TextManager(string={'constant': 'A'})
features = pd.DataFrame(index=range(5))
text_manager.apply(features)
np.testing.assert_array_equal(text_manager.values, 'A')
def test_apply_with_constant_color():
color = {'constant': 'red'}
features = pd.DataFrame({'class': ['A', 'B', 'C']})
text_manager = TextManager(color=color, features=features)
features = pd.DataFrame({'class': ['A', 'B', 'C', 'D', 'E']})
text_manager.apply(features)
actual = text_manager.color._values
assert_colors_equal(actual, 'red')
def test_apply_with_derived_color():
color = {'feature': 'colors'}
features = pd.DataFrame({'colors': ['red', 'green', 'blue']})
text_manager = TextManager(color=color, features=features)
features = pd.DataFrame(
{'colors': ['red', 'green', 'blue', 'yellow', 'cyan']})
text_manager.apply(features)
actual = text_manager.color._values
assert_colors_equal(actual, ['red', 'green', 'blue', 'yellow', 'cyan'])
def test_apply_with_derived_string():
features = pd.DataFrame({'class': ['A', 'B', 'C']})
text_manager = TextManager(string='class: {class}', features=features)
features = pd.DataFrame({'class': ['A', 'B', 'C', 'D', 'E']})
text_manager.apply(features)
np.testing.assert_array_equal(
text_manager.values,
['class: A', 'class: B', 'class: C', 'class: D', 'class: E'],
)
def test_apply_with_manual_color_then_use_default():
color = {
'array': ['red', 'green', 'blue'],
'default': 'yellow',
}
features = pd.DataFrame({'class': ['A', 'B', 'C']})
text_manager = TextManager(color=color, features=features)
features = pd.DataFrame({'class': ['A', 'B', 'C', 'D', 'E']})
text_manager.apply(features)
actual = text_manager.color._values
assert_colors_equal(actual, ['red', 'green', 'blue', 'yellow', 'yellow'])
def test_apply_with_manual_string():
string = {
'array': ['A', 'B', 'C'],
'default': 'D',
}
features = pd.DataFrame(index=range(3))
text_manager = TextManager(string=string, features=features)
features = pd.DataFrame(index=range(5))
text_manager.apply(features)
np.testing.assert_array_equal(text_manager.values,
['A', 'B', 'C', 'D', 'D'])
