lucuma.catalog.arb.ArbBrightnessConstraints.scala Maven / Gradle / Ivy
// Copyright (c) 2016-2023 Association of Universities for Research in Astronomy, Inc. (AURA)
// For license information see LICENSE or https://opensource.org/licenses/BSD-3-Clause
package lucuma.catalog.arb
import lucuma.catalog.BandsList
import lucuma.catalog.BrightnessConstraints
import lucuma.catalog.FaintnessConstraint
import lucuma.catalog.SaturationConstraint
import lucuma.core.math.BrightnessValue
import lucuma.core.math.arb.ArbBrightnessValue.given
import org.scalacheck.*
import org.scalacheck.Arbitrary.arbitrary
trait ArbBrightnessConstraints {
given Arbitrary[FaintnessConstraint] =
Arbitrary {
for {
b <- arbitrary[BrightnessValue]
} yield FaintnessConstraint(b)
}
given Cogen[FaintnessConstraint] =
Cogen[BrightnessValue].contramap(_.brightness)
given Arbitrary[SaturationConstraint] =
Arbitrary {
for {
b <- arbitrary[BrightnessValue]
} yield SaturationConstraint(b)
}
given Cogen[SaturationConstraint] =
Cogen[BrightnessValue].contramap(_.brightness)
given Arbitrary[BrightnessConstraints] =
Arbitrary {
for {
f <- arbitrary[FaintnessConstraint]
l <- arbitrary[Option[SaturationConstraint]]
} yield BrightnessConstraints(BandsList.GaiaBandsList, f, l)
}
given Cogen[BrightnessConstraints] =
Cogen[(FaintnessConstraint, Option[SaturationConstraint])].contramap(x =>
(x.faintnessConstraint, x.saturationConstraint)
)
}
object ArbBrightnessConstraints extends ArbBrightnessConstraints