mat_test.go

// Copyright 2022 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package mat_test

import (
	"fmt"
	"github.com/dr2chase/mat"
	"testing"
)

func rmAndCm() (rm, cm mat.MuM[mat.F]) {
	rm = mat.RowMajor[mat.F](5, 5)
	cm = mat.ColumnMajor[mat.F](5, 5)
	f := func(i, j int) mat.F {
		diff := j - i
		switch diff {
		case 0:
			return mat.F(1)
		case -1, 1:
			return mat.F(2 * diff)
		}
		return mat.F(0)
	}
	mat.SetByIJ[mat.F](rm, f)
	mat.SetByIJ[mat.F](cm, f)
	return
}

func TestPrint(t *testing.T) {
	rm, cm := rmAndCm()
	fmt.Println("Row major matrix")
	rm.Print()
	fmt.Println("Column major matrix")
	cm.Print()
}

func TestEqual(t *testing.T) {
	rm, cm := rmAndCm()
	if !mat.EqualsM(mat.M[mat.F](rm), mat.M[mat.F](cm)) {
		t.Fail()
	}
	if !cm.Equals(rm) {
		t.Fail()
	}
}

func TestV(t *testing.T) {
	// rm, cm := rmAndCm()
	v := mat.Vector[mat.F](5)
	v.SetByI(func(i int) mat.F { return mat.F(i + 1) })
	v.Print()
}

func TestVM(t *testing.T) {
	rm, cm := rmAndCm()
	v := mat.Vector[mat.F](5)
	v.SetByI(func(i int) mat.F { return mat.F(i + 1) })
	// v.Print()
	rmv := rm.TimesVector(v)
	rmv.Print()
	cmv := cm.TimesVector(v)
	cmv.Print()
	rmvl := rm.LeftTimesVector(v)
	rmvl.Print()
	cmvl := cm.LeftTimesVector(v)
	cmvl.Print()
	rmtvl := rm.Transpose().LeftTimesVector(v)
	rmtvl.Print()

	if !rmv.Equals(cmv) {
		t.Fail()
	}
	if !rmvl.Equals(cmvl) {
		t.Fail()
	}
	if !rmtvl.Equals(cmv) {
		t.Fail()
	}
	fmt.Println("rmv rmv = ", rmv.Inner(rmv))
	fmt.Println("rmvl rmvl = ", rmvl.Inner(rmvl))
	fmt.Println("rmv rmvl = ", rmv.Inner(rmvl))
}