add unit test

4 years ago · eb4d3afbea
parent 2c4e279f1e
commit eb4d3afbea
1 changed files with 14 additions and 0 deletions
--- a/str/str_chr.c
+++ b/str/str_chr.c
@ -15,3 +15,17 @@ size_t str_chr(const char *in, char needle) {
  }
  return (size_t)(t-in);
 }
+
+#ifdef UNITTEST
+#include <assert.h>
+#include <string.h>
+
+int main() {
+  char buf[100];
+  strcpy(buf,"abcdefghijklmnopqrstuvwxyz");
+  size_t i;
+  for (i=0; i<26; ++i)
+    assert(str_chr(buf,buf[i])==i);
+  assert(str_chr(buf,'X')==26);
+}
+#endif