package autotrader

import (
	"testing"
)

func TestRSI(t *testing.T) {
	prices := NewDataSeriesFloat("Prices", 1, 0, 2, 1, 3, 2, 4, 3, 5, 4, 6, 5, 7, 6)
	rsi := RSI(prices, 14)
	if rsi.Len() != 14 {
		t.Errorf("RSI length is %d, expected 14", rsi.Len())
	}
	if !EqualApprox(rsi.Float(0), 100) {
		t.Errorf("RSI[0] is %f, expected 0", rsi.Float(0))
	}
	if !EqualApprox(rsi.Float(-1), 61.02423) {
		t.Errorf("RSI[-1] is %f, expected 100", rsi.Float(-1))
	}
}