Add bytearray.{lower, upper}

Author: janczer

tests/snippets/bytearray.py

@@ -18,13 +18,19 @@
 assert bytearray(b'1234567890').isdigit()
 assert not bytearray(b'12ab').isdigit()
 
-assert bytearray(b'lower').islower()
+l = bytearray(b'lower')
+assert l.islower()
+assert not l.isupper()
+assert l.upper().isupper()
 assert not bytearray(b'Super Friends').islower()
 
 assert bytearray(b' \n\t').isspace()
 assert not bytearray(b'\td\n').isspace()
 
-assert bytearray(b'UPPER').isupper()
+b = bytearray(b'UPPER')
+assert b.isupper()
+assert not b.islower()
+assert b.lower().islower()
 assert not bytearray(b'tuPpEr').isupper()
 
 assert bytearray(b'Is Title Case').istitle()

vm/src/obj/objbytearray.rs

@@ -102,6 +102,16 @@ pub fn init(context: &PyContext) {
         context.new_rustfunc(bytearray_clear),
     );
     context.set_attr(&bytearray_type, "pop", context.new_rustfunc(bytearray_pop));
+    context.set_attr(
+        &bytearray_type,
+        "lower",
+        context.new_rustfunc(bytearray_lower),
+    );
+    context.set_attr(
+        &bytearray_type,
+        "upper",
+        context.new_rustfunc(bytearray_upper),
+    );
 }
 
 fn bytearray_new(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
@@ -188,7 +198,7 @@ fn bytearray_islower(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
         !bytes.is_empty()
             && bytes
                 .iter()
-                .filter(|x| char::from(**x).is_whitespace())
+                .filter(|x| !char::from(**x).is_whitespace())
                 .all(|x| char::from(*x).is_lowercase()),
     ))
 }
@@ -295,3 +305,21 @@ fn bytearray_pop(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
         Err(vm.new_index_error("pop from empty bytearray".to_string()))
     }
 }
+
+fn bytearray_lower(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args, required = [(obj, Some(vm.ctx.bytearray_type()))]);
+    let value = get_value(obj).to_vec().to_ascii_lowercase();
+    Ok(PyObject::new(
+        PyObjectPayload::Bytes { value },
+        vm.ctx.bytearray_type(),
+    ))
+}
+
+fn bytearray_upper(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args, required = [(obj, Some(vm.ctx.bytearray_type()))]);
+    let value = get_value(obj).to_vec().to_ascii_uppercase();
+    Ok(PyObject::new(
+        PyObjectPayload::Bytes { value },
+        vm.ctx.bytearray_type(),
+    ))
+}