def test_format_text_properties_no_matching_props(): text = 'hi' n_text = 5 formatted_text, text_mode = format_text_properties(text, n_text=5) expected_text = np.repeat(text, n_text) np.testing.assert_equal(formatted_text, expected_text) properties = {'hello': [1, 2, 3, 4, 5]} formatted_text, text_mode = format_text_properties(text, n_text=5, properties=properties) expected_text = np.repeat(text, n_text) np.testing.assert_equal(formatted_text, expected_text)
def test_format_text_properties(): text = 'name' n_text = 2 properties = { 'name': np.array(['bob', 'jane']), 'item': np.array(['ducks', 'microscopes']), 'weight': np.array([3, 15.123]), } formatted_text, text_mode = format_text_properties(text, n_text, properties) expected_text = ['bob', 'jane'] np.testing.assert_equal(formatted_text, expected_text) assert isinstance(formatted_text, np.ndarray)
def test_format_text_properties_f_string(): text = '{name} has 5 {item} that weigh {weight:.1f} kgs' n_text = 2 properties = { 'name': np.array(['bob', 'jane']), 'item': np.array(['ducks', 'microscopes']), 'weight': np.array([3, 15.123]), } formatted_text, text_mode = format_text_properties(text, n_text, properties) expected_text = [ 'bob has 5 ducks that weigh 3.0 kgs', 'jane has 5 microscopes that weigh 15.1 kgs', ] np.testing.assert_equal(formatted_text, expected_text) assert isinstance(formatted_text, np.ndarray)