make special editable as well

This commit is contained in:
Arthur Zucker
2023-09-04 20:54:29 +00:00
parent 2291c89896
commit 058e34b421
3 changed files with 14 additions and 0 deletions

View File

@ -216,6 +216,12 @@ impl PyAddedToken {
self.get_token().special
}
/// Set the value of the :obj:`special` option
#[setter]
fn set_special(&mut self, special: bool) {
self.special = special;
}
fn __str__(&self) -> PyResult<&str> {
Ok(&self.content)
}

View File

@ -20,6 +20,10 @@ class TestAddedToken:
assert added_token.content == "<MASK>"
assert type(added_token) == AddedToken
added_token.content = added_token.content.lower()
assert added_token.special == False
added_token.special = True
assert added_token.special == True
assert str(added_token) == "<mask>"
assert (

View File

@ -678,6 +678,10 @@ mod tests {
let mut token: AddedToken = AddedToken::from("Hey", false);
token.content = "hey".to_string();
assert_eq!(token.content, "hey"); // Token was already there
token.special = true;
assert_eq!(token.special, true); // Token was already there
}
#[test]