map: fixed deletion of non-existent keys

pull/2593/head
Amal Shaji 2019-10-30 22:49:59 +05:30 committed by Alexander Medvednikov
parent 5da3c49497
commit e8d34002e1
2 changed files with 6 additions and 2 deletions

View File

@ -207,8 +207,10 @@ pub fn (n mut mapnode) delete(key string, element_size int) {
} }
pub fn (m mut map) delete(key string) { pub fn (m mut map) delete(key string) {
if m.exists(key) {
m.root.delete(key, m.element_size) m.root.delete(key, m.element_size)
m.size-- m.size--
}
} }
fn (m map) exists(key string) bool { fn (m map) exists(key string) bool {

View File

@ -38,6 +38,8 @@ fn test_map() {
assert keys[1] == 'hello' assert keys[1] == 'hello'
m.delete('hi') m.delete('hi')
assert m.size == 1 assert m.size == 1
m.delete('aloha')
assert m.size == 1
assert m['hi'] == 0 assert m['hi'] == 0
assert m.keys().len == 1 assert m.keys().len == 1