zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 04:24:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adds radians and degrees (#1011)

Browse Source
This commit is contained in:
Aneesh Shetty
2024-04-22 13:17:49 -05:00
committed by GitHub
parent 3d405fb3b1
commit d0dbfe0b97
7 changed files with 120 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
tests/ops_tests.cpp
View File

@@ -1154,6 +1154,56 @@ TEST_CASE("test arithmetic unary ops") {
CHECK(allclose(cos(x), expected).item<bool>());
}
// Test degrees
{
array x(0.0);
CHECK_EQ(degrees(x).item<float>(), 0.0);
x = array(M_PI_2);
CHECK(degrees(x).item<float>() == doctest::Approx(90.0));
CHECK(array_equal(degrees(array({})), array({})).item<bool>());
// Integer input type
x = array(0);
CHECK_EQ(x.dtype(), int32);
CHECK_EQ(degrees(x).item<float>(), 0.0);
// Input is irregularly strided
x = broadcast_to(array(M_PI_2), {2, 2, 2});
CHECK(allclose(degrees(x), full({2, 2, 2}, 90.0)).item<bool>());
float angles[] = {0.0f, M_PI_2, M_PI, 3.0f * M_PI_2};
x = split(array(angles, {2, 2}), 2, 1)[0];
auto expected = array({0.0f, 180.0f}, {2, 1});
CHECK(allclose(degrees(x), expected).item<bool>());
}
// Test radians
{
array x(0.0);
CHECK_EQ(radians(x).item<float>(), 0.0);
x = array(90.0);
CHECK(radians(x).item<float>() == doctest::Approx(M_PI_2));
CHECK(array_equal(radians(array({})), array({})).item<bool>());
// Integer input type
x = array(90);
CHECK_EQ(x.dtype(), int32);
CHECK(radians(x).item<float>() == doctest::Approx(M_PI_2));
// Input is irregularly strided
x = broadcast_to(array(90.0f), {2, 2, 2});
CHECK(allclose(radians(x), full({2, 2, 2}, M_PI_2)).item<bool>());
x = split(array({0.0f, 90.0f, 180.0f, 270.0f}, {2, 2}), 2, 1)[0];
float angles[] = {0.0f, M_PI};
auto expected = array(angles, {2, 1});
CHECK(allclose(radians(x), expected).item<bool>());
}
// Test log
{
array x(0.0);