mirror of
https://github.com/ncruces/go-sqlite3.git
synced 2026-01-12 05:59:14 +00:00
56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
package stats_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/ncruces/go-sqlite3"
|
|
_ "github.com/ncruces/go-sqlite3/embed"
|
|
"github.com/ncruces/go-sqlite3/ext/stats"
|
|
_ "github.com/ncruces/go-sqlite3/tests/testcfg"
|
|
)
|
|
|
|
func TestRegister_quantile(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db, err := sqlite3.Open(":memory:")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
stats.Register(db)
|
|
|
|
err = db.Exec(`CREATE TABLE data (x)`)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = db.Exec(`INSERT INTO data (x) VALUES (4), (7.0), ('13'), (NULL), (16)`)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
stmt, _, err := db.Prepare(`
|
|
SELECT
|
|
median(x),
|
|
quantile_disc(x, 0.5),
|
|
quantile_cont(x, 0.3)
|
|
FROM data`)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer stmt.Close()
|
|
|
|
if stmt.Step() {
|
|
if got := stmt.ColumnFloat(0); got != 10 {
|
|
t.Errorf("got %v, want 10", got)
|
|
}
|
|
if got := stmt.ColumnFloat(1); got != 7 {
|
|
t.Errorf("got %v, want 7", got)
|
|
}
|
|
if got := stmt.ColumnFloat(2); got != 6.699999999999999 {
|
|
t.Errorf("got %v, want 6.7", got)
|
|
}
|
|
}
|
|
}
|