/****************************************************************************
 * Copyright (c) 2025, ArborX authors                                       *
 * All rights reserved.                                                     *
 *                                                                          *
 * This file is part of the ArborX library. ArborX is                       *
 * distributed under a BSD 3-clause license. For the licensing terms see    *
 * the LICENSE file in the top-level directory.                             *
 *                                                                          *
 * SPDX-License-Identifier: BSD-3-Clause                                    *
 ****************************************************************************/

#include <ArborX_MinimumSpanningTree.hpp>

namespace
{

KOKKOS_FUNCTION constexpr bool test_directed_edges()
{
  using namespace ArborX::Details;

  static_assert(DirectedEdge{0, 1, 2.f}.weight == 2.f);
  static_assert(DirectedEdge{0, 1, 2.f}.source() == 0);
  static_assert(DirectedEdge{0, 1, 2.f}.target() == 1);

  static_assert(DirectedEdge{6, 5, 4.f}.weight == 4.f);
  static_assert(DirectedEdge{6, 5, 4.f}.source() == 6);
  static_assert(DirectedEdge{6, 5, 4.f}.target() == 5);

  constexpr auto inf = KokkosExt::ArithmeticTraits::infinity<float>::value;
  static_assert(DirectedEdge{}.weight == inf);
  static_assert(DirectedEdge{}.source() == INT_MAX);
  static_assert(DirectedEdge{}.target() == INT_MAX);

  static_assert(DirectedEdge{0, 1, 2.f} < DirectedEdge{0, 1, 3.f});
  static_assert(DirectedEdge{0, 1, 2.f} < DirectedEdge{1, 2, 2.f});
  static_assert(DirectedEdge{0, 1, 2.f} < DirectedEdge{2, 1, 2.f});

  constexpr DirectedEdge e12{1, 2, 3};
  constexpr DirectedEdge e21{2, 1, 3};
  static_assert(e12.weight == e21.weight);
  static_assert((e12 < e21) ^ (e21 < e12));

  return true;
}
static_assert(test_directed_edges()); // avoid warning unused function

KOKKOS_FUNCTION constexpr bool test_weighted_edges()
{
  using ArborX::Experimental::WeightedEdge;

  static_assert(WeightedEdge{1, 2, 3} < WeightedEdge{1, 2, 4});

  constexpr WeightedEdge e12{1, 2, 3};
  constexpr WeightedEdge e21{2, 1, 3};
  static_assert(e12.weight == e21.weight);
  static_assert(!(e12 < e21));
  static_assert(!(e21 < e12));

  constexpr WeightedEdge e14{1, 4, 3};
  static_assert(e12.weight == e14.weight);
  static_assert(e12 < e14);

  return true;
}
static_assert(test_weighted_edges()); // avoid warning unused function

} // namespace