ssa

1
package pmtm_test
2

3
import (
4
	"fmt"
5
	"strconv"
6
	"testing"
7

8
	"github.com/RB-PRO/ssa/pkg/oss"
9
	"github.com/RB-PRO/ssa/pkg/pmtm"
10
	"github.com/xuri/excelize/v2"
11
)
12

13
func TestPmtm(t *testing.T) {
14
	x, _ := oss.Make_singnal_xn("pmtm")
15
	y := pmtm.Pmtm(x, 1024)
16

17
	safeToXlsx(x, y)
18
}
19

20
// Сохранить в xlsx для дебага
21
func safeToXlsx(x, y []float64) {
22

23
	file_graph := excelize.NewFile()
24
	file_graph.NewSheet("main")
25
	file_graph.DeleteSheet("Sheet1")
26
	lenFor := len(x)
27
	for ind := 0; ind < lenFor; ind++ {
28
		file_graph.SetCellValue("golang", "A"+strconv.Itoa(ind+1), x[ind])
29
	}
30
	lenFor = len(y)
31
	for ind := 0; ind < lenFor; ind++ {
32
		file_graph.SetCellValue("golang", "B"+strconv.Itoa(ind+1), y[ind])
33
	}
34
	if err := file_graph.SaveAs("save_pmtm" + ".xlsx"); err != nil {
35
		fmt.Println(err)
36
	}
37
}
38