Implement endswith() method

Author: atpalmer

README.md

@@ -71,6 +71,16 @@ Notes:
 * `start` and `end`, if provided, are _byte_ indexes.
 
 
+### endswith(substring [,start [,end]])
+
+See: https://docs.python.org/3/library/stdtypes.html#str.endswith
+
+Notes:
+
+* `substring` may be a `cstring` or Python `str` object.
+* `start` and `end`, if provided, are _byte_ indexes.
+
+
 ## TODO
 
 * Write docs (see `str` type docs)

src/cstring.c

@@ -344,6 +344,17 @@ PyObject *cstring_startswith(PyObject *self, PyObject *args) {
     return PyBool_FromLong(cmp == 0);
 }
 
+PyDoc_STRVAR(endswith__doc__, "");
+PyObject *cstring_endswith(PyObject *self, PyObject *args) {
+    struct _substr_params params;
+    if(!_parse_substr_args(self, args, &params))
+        return NULL;
+    if(params.end - params.start < params.substr_len)
+        return PyBool_FromLong(0);
+    int cmp = memcmp(params.end - params.substr_len, params.substr, params.substr_len);
+    return PyBool_FromLong(cmp == 0);
+}
+
 static PySequenceMethods cstring_as_sequence = {
     .sq_length = cstring_len,
     .sq_concat = cstring_concat,
@@ -364,6 +375,7 @@ static PyMethodDef cstring_methods[] = {
     {"rfind", cstring_rfind, METH_VARARGS, rfind__doc__},
     {"rindex", cstring_rindex, METH_VARARGS, rindex__doc__},
     {"startswith", cstring_startswith, METH_VARARGS, startswith__doc__},
+    {"endswith", cstring_endswith, METH_VARARGS, endswith__doc__},
     {0},
 };
 

test/test_methods.py

@@ -113,3 +113,23 @@ def test_startswith_with_start_and_end():
     target = cstring('hello, world')
     assert target.startswith('wo', 7, 8) is False
 
+
+def test_endswith():
+    target = cstring('hello, world')
+    assert target.endswith('world') is True
+
+
+def test_endswith_not():
+    target = cstring('hello, world')
+    assert target.endswith('hello') is False
+
+
+def test_endswith_with_start():
+    target = cstring('hello, world')
+    assert target.endswith('world', 8) is False
+
+
+def test_endswith_with_start_and_end():
+    target = cstring('hello, world')
+    assert target.endswith('wo', 7, 9) is True
+