golearn/base/lazy_sort_test.go

package base

import (
	"testing"
)

func TestLazySortDesc(t *testing.T) {
	inst1, err := ParseCSVToInstances("../examples/datasets/iris_headers.csv", true)
	if err != nil {
		t.Error(err)
		return
	}
	inst2, err := ParseCSVToInstances("../examples/datasets/iris_sorted_desc.csv", true)
	if err != nil {
		t.Error(err)
		return
	}

	as1 := ResolveAllAttributes(inst1)
	as2 := ResolveAllAttributes(inst2)

	if isSortedDesc(inst1, as1[0]) {
		t.Error("Can't test descending sort order")
	}
	if !isSortedDesc(inst2, as2[0]) {
		t.Error("Reference data not sorted in descending order!")
	}

	inst, err := LazySort(inst1, Descending, as1[0:len(as1)-1])
	if err != nil {
		t.Error(err)
	}
	if !isSortedDesc(inst, as1[0]) {
		t.Error("Instances are not sorted in descending order")
		t.Error(inst1)
	}
	if !inst2.Equal(inst) {
		t.Error("Instances don't match")
		t.Error(inst)
		t.Error(inst2)
	}
}

func TestLazySortAsc(t *testing.T) {
	inst, err := ParseCSVToInstances("../examples/datasets/iris_headers.csv", true)
	as1 := ResolveAllAttributes(inst)
	if isSortedAsc(inst, as1[0]) {
		t.Error("Can't test ascending sort on something ascending already")
	}
	if err != nil {
		t.Error(err)
		return
	}
	insts, err := LazySort(inst, Ascending, as1)
	if err != nil {
		t.Error(err)
		return
	}
	if !isSortedAsc(insts, as1[0]) {
		t.Error("Instances are not sorted in ascending order")
		t.Error(insts)
	}

	inst2, err := ParseCSVToInstances("../examples/datasets/iris_sorted_asc.csv", true)
	if err != nil {
		t.Error(err)
		return
	}
	as2 := ResolveAllAttributes(inst2)
	if !isSortedAsc(inst2, as2[0]) {
		t.Error("This file should be sorted in ascending order")
	}

	if !inst2.Equal(insts) {
		t.Error("Instances don't match")
		t.Error(inst)
		t.Error(inst2)
	}

	rowStr := insts.RowString(0)
	ref := "4.30 3.00 1.10 0.10 Iris-setosa"
	if rowStr != ref {
		t.Fatalf("'%s' != '%s'", rowStr, ref)
	}

}