fn_clamp.cpp 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. // Copyright 2015 Conrad Sanderson (http://conradsanderson.id.au)
  2. // Copyright 2015 National ICT Australia (NICTA)
  3. //
  4. // Licensed under the Apache License, Version 2.0 (the "License");
  5. // you may not use this file except in compliance with the License.
  6. // You may obtain a copy of the License at
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. // ------------------------------------------------------------------------
  15. #include <armadillo>
  16. #include "catch.hpp"
  17. using namespace arma;
  18. TEST_CASE("fn_clamp_1")
  19. {
  20. mat A = randu<mat>(5,6);
  21. mat B = clamp(A, 0.2, 0.8);
  22. REQUIRE( B.min() == Approx(0.2) );
  23. REQUIRE( B.max() == Approx(0.8) );
  24. mat C = clamp(A, A.min(), 0.8);
  25. REQUIRE( C.min() == A.min() );
  26. REQUIRE( C.max() == Approx(0.8) );
  27. mat D = clamp(A, 0.2, A.max());
  28. REQUIRE( D.min() == Approx(0.2) );
  29. REQUIRE( D.max() == A.max() );
  30. REQUIRE_THROWS( clamp(A, A.max(), A.min() ) );
  31. }