Test case and documentation update for suffx and prefix functionality

README.md

@@ -48,10 +48,15 @@ Convert string to camel case, snake case, kebab case / slugify, custom delimiter
         <td><a href="#ucfirst-string">UcFirst</a></td>
     </tr>
      <tr>
-            <td><a href="#firstlength-int-string">First</a></td>
+         <td><a href="#firstlength-int-string">First</a></td>
-            <td><a href="#lastlength-int-string">Last</a></td>
+         <td><a href="#lastlength-int-string">Last</a></td>
-            <td></td>
+         <td><a href="#prefix-string-string">Prefix</a></td>
-        </tr>
+     </tr>
+     <tr>
+         <td><a href="#suffix-string-string">Suffix</a></td>
+         <td></td>
+         <td></td>
+     </tr>
 </table>
 
 

stringy_test.go

@@ -1,7 +1,6 @@
 package stringy
 
 import (
-	"fmt"
 	"testing"
 )
 
@@ -312,28 +311,30 @@ func TestInput_Last(t *testing.T) {
 	}
 }
 
-func ExampleStringManipulation_Prefix() {
+func TestInput_Prefix(t *testing.T) {
-	fmt.Println(New("foobar").Prefix("foo"))
+	str := New("foobar")
-	fmt.Println(New("foobar").Prefix("foofoo"))
+	against := "foobar"
-	fmt.Println(New("foobar").Prefix("/"))
+	if val := str.Prefix("foo"); val != against {
-	fmt.Println(New("").Prefix("_"))
+		t.Errorf("Expected: to be %s but got: %s", against, val)
-	// Output:
+	}
-	// foobar
+
-	// foofoofoobar
+	str = New("foobar")
-	// /foobar
+	against = "foofoofoobar"
-	// _
+	if val := str.Prefix("foofoo"); val != against {
+		t.Errorf("Expected: to be %s but got: %s", against, val)
+	}
 }
 
-func ExampleStringManipulation_Suffix() {
+func TestInput_Suffix(t *testing.T) {
-	fmt.Println(New("foobar").Suffix("bar"))
+	str := New("foobar")
-	fmt.Println(New("foobar").Suffix("barbar"))
+	against := "foobar"
-	fmt.Println(New("foobar").Suffix("/"))
+	if val := str.Suffix("bar"); val != against {
-	fmt.Println(New("foobar/").Suffix("/"))
+		t.Errorf("Expected: to be %s but got: %s", against, val)
-	fmt.Println(New("").Suffix("_"))
+	}
-	// Output:
+
-	// foobar
+	str = New("foobar")
-	// foobarbarbar
+	against = "foobarbarbar"
-	// foobar/
+	if val := str.Suffix("barbar"); val != against {
-	// foobar/
+		t.Errorf("Expected: to be %s but got: %s", against, val)
-	// _
+	}
 }