Add test case to check image type

imgtype.go

@@ -3,11 +3,12 @@ package imgtype // import "github.com/shamsher31/goimgtype"
 
 import (
 	"errors"
-	"github.com/shamsher31/goimgext"
 	"log"
 	"net/http"
 	"os"
 	"strings"
+
+	"github.com/shamsher31/goimgext"
 )
 
 // Get returns the type of Image

imgtype_test.go

 package imgtype
 
-import "testing"
+import (
+	"strings"
+	"testing"
+
+	"github.com/shamsher31/goimgext"
+)
 
 func TestGet(t *testing.T) {
+	imgType, err := Get("golang.png")
+
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if isImgTypeFromExt(imgType) == false {
+		t.Fatal("Not a valid image type")
+	}
+}
 
+func isImgTypeFromExt(p string) bool {
+	ext := imgext.Get()
+	for i := 0; i < len(ext); i++ {
+		if strings.Contains(ext[i], p[6:len(p)]) {
+			return true
+		}
+	}
+	return false
 }