package iskra

import "testing"

func TestBigramKey(t *testing.T) {
	k1 := BigramKey(StageSRC, []byte("func"), []byte("main"))
	k2 := BigramKey(StageSRC, []byte("func"), []byte("main"))
	if k1 != k2 {
		t.Fatal("same bigram should produce same key")
	}

	k3 := BigramKey(StageSRC, []byte("main"), []byte("func"))
	if k1 == k3 {
		t.Fatal("reversed bigram should produce different key")
	}

	k4 := BigramKey(StageIR, []byte("func"), []byte("main"))
	if k1 == k4 {
		t.Fatal("different stage should produce different key")
	}
}