Test script should be a .zsh file instead of .sh

script/test.zsh

@@ -0,0 +1,77 @@
+#!/usr/bin/env zsh
+
+SCRIPT_DIR=$(dirname "$0")
+TEST_DIR=$SCRIPT_DIR/../test
+DIST_DIR=$SCRIPT_DIR/../dist
+
+source $DIST_DIR/autosuggestions.zsh
+
+testDefaultHighlightStyle() {
+	assertEquals \
+		"fg=8" \
+		"$ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE"
+}
+
+testHighlightApplyWithSuggestion() {
+	orig_style=ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE
+	ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE="fg=4"
+
+	BUFFER="ec"
+	POSTDISPLAY="ho hello"
+	region_highlight=("0 2 fg=1")
+
+	_zsh_autosuggest_highlight_apply
+
+	assertEquals \
+		"adds to region_highlight with correct style" \
+		"0 2 fg=1 2 10 $ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE" \
+		"$region_highlight"
+
+	assertEquals \
+		"saves the higlight to be removed later" \
+		"2 10 $ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE" \
+		"$_ZSH_AUTOSUGGEST_LAST_HIGHLIGHT"
+
+	ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE=orig_style
+}
+
+testHighlightApplyWithoutSuggestion() {
+	BUFFER="echo hello"
+	POSTDISPLAY=""
+	region_highlight=("0 4 fg=1")
+
+	_zsh_autosuggest_highlight_apply
+
+	assertEquals \
+		"leaves region_highlight alone" \
+		"0 4 fg=1" \
+		"$region_highlight"
+
+	assertNull \
+		"clears the last highlight" \
+		"$_ZSH_AUTOSUGGEST_LAST_HIGHLIGHT"
+}
+
+testHighlightReset() {
+	BUFFER="ec"
+	POSTDISPLAY="ho hello"
+	region_highlight=("0 1 fg=1" "2 10 fg=8" "1 2 fg=1")
+	_ZSH_AUTOSUGGEST_LAST_HIGHLIGHT="2 10 fg=8"
+
+	_zsh_autosuggest_highlight_reset
+
+	assertEquals \
+		"removes last highlight region" \
+		"0 1 fg=1 1 2 fg=1" \
+		"$region_highlight"
+
+	assertNull \
+		"clears the last highlight" \
+		"$_ZSH_AUTOSUGGEST_LAST_HIGHLIGHT"
+}
+
+# For zsh compatibility
+setopt shwordsplit
+SHUNIT_PARENT=$0
+
+source $TEST_DIR/shunit2-2.1.6/src/shunit2