lucuma.catalog.arb.ArbBrightnessConstraints.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lucuma-catalog-testkit_3 Show documentation

lucuma-catalog-testkit

There is a newer version: 0.48.12

Show newest version

// 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