32 lines
1.1 KiB
Go
32 lines
1.1 KiB
Go
|
// Package ephemtest 测试辅助：固定星历模型、按模型选择基线文件。
|
||
|
|
package ephemtest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"b612.me/astro/internal/ephem"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Pin 在本测试内固定模型，结束后恢复 / pins the model for the duration of a test.
|
||
|
|
func Pin(tb testing.TB, m ephem.Model) {
|
||
|
|
prev := ephem.Active()
|
||
|
|
ephem.SetActive(m)
|
||
|
|
tb.Cleanup(func() { ephem.SetActive(prev) })
|
||
|
|
}
|
||
|
|
|
||
|
|
// PinLegacy 只针对旧级数实现本身的测试使用 / for tests of the legacy series implementation itself.
|
||
|
|
func PinLegacy(tb testing.TB) { Pin(tb, ephem.Legacy) }
|
||
|
|
|
||
|
|
// Golden 当前模型对应的基线文件：旧模型用原路径，新模型用同目录 semianalytic/ 下的同名文件。
|
||
|
|
func Golden(path string) string {
|
||
|
|
if ephem.Active() != ephem.Semianalytic {
|
||
|
|
return path
|
||
|
|
}
|
||
|
|
return filepath.Join(filepath.Dir(path), "semianalytic", filepath.Base(path))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update 环境变量 ASTRO_UPDATE_GOLDEN=1 时重写当前模型的基线 / whether to rewrite the active model's baselines.
|
||
|
|
func Update() bool { return os.Getenv("ASTRO_UPDATE_GOLDEN") == "1" }
|