def test_refresh_with_manual_string():
features = pd.DataFrame(index=range(3))
text_manager = TextManager(string=['A', 'B', 'C'], features=features)
text_manager.string = ['C', 'B', 'A']
text_manager.refresh(features)
np.testing.assert_array_equal(text_manager.values, ['C', 'B', 'A'])
def test_refresh_with_constant_string():
features = pd.DataFrame(index=range(3))
text_manager = TextManager(string={'constant': 'A'})
text_manager.string = {'constant': 'B'}
text_manager.refresh(features)
np.testing.assert_array_equal(text_manager.values, 'B')
def test_refresh_with_derived_color():
color = {'feature': 'colors'}
features = pd.DataFrame({'colors': ['red', 'green', 'blue']})
text_manager = TextManager(color=color, features=features)
features = pd.DataFrame({'colors': ['green', 'yellow', 'magenta']})
text_manager.refresh(features)
actual = text_manager.color._values
assert_colors_equal(actual, ['green', 'yellow', 'magenta'])
def test_refresh_with_manual_color():
color = ['red', 'green', 'blue']
features = pd.DataFrame(index=range(3))
text_manager = TextManager(color=color, features=features)
text_manager.color = ['green', 'cyan', 'yellow']
text_manager.refresh(features)
actual = text_manager.color._values
assert_colors_equal(actual, ['green', 'cyan', 'yellow'])
def test_refresh_with_constant_color():
color = {'constant': 'red'}
features = pd.DataFrame(index=range(3))
text_manager = TextManager(color=color, features=features)
text_manager.color = {'constant': 'yellow'}
text_manager.refresh(features)
actual = text_manager.color._values
assert_colors_equal(actual, 'yellow')
def test_refresh_with_derived_string():
features = pd.DataFrame({'class': ['A', 'B', 'C']})
text_manager = TextManager(string='class: {class}', features=features)
features = pd.DataFrame({'class': ['E', 'D', 'C', 'B', 'A']})
text_manager.refresh(features)
np.testing.assert_array_equal(
text_manager.values,
['class: E', 'class: D', 'class: C', 'class: B', 'class: A'],
)
