PikaMath-test.cpp 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292
  1. #include "test_common.h"
  2. TEST_START
  3. #if PIKA_SYNTAX_LEVEL == PIKA_SYNTAX_LEVEL_MAXIMAL
  4. TEST(PikaMath, test1) {
  5. /* init */
  6. g_PikaMemInfo.heapUsedMax = 0;
  7. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  8. /* run */
  9. __platform_printf("BEGIN\r\n");
  10. pikaVM_runSingleFile(pikaMain, "test/python/PikaMath/Quaternion_test.py");
  11. /* collect */
  12. /* assert */
  13. /* deinit */
  14. obj_deinit(pikaMain);
  15. EXPECT_EQ(pikaMemNow(), 0);
  16. }
  17. TEST(PikaMath, quat_add) {
  18. /* init */
  19. g_PikaMemInfo.heapUsedMax = 0;
  20. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  21. /* run */
  22. obj_run(pikaMain,
  23. "import PikaMath\n"
  24. "a = PikaMath.Quaternion()\n"
  25. "b = PikaMath.Quaternion()\n"
  26. "a.add(b)\n"
  27. "res1=a.get(0)\n"
  28. "res2=a.get(1)\n"
  29. "res3=a.get(2)\n"
  30. "res4=a.get(3)\n");
  31. /* collect */
  32. float res1 = obj_getFloat(pikaMain, "res1");
  33. float res2 = obj_getFloat(pikaMain, "res2");
  34. float res3 = obj_getFloat(pikaMain, "res3");
  35. float res4 = obj_getFloat(pikaMain, "res4");
  36. /* assert */
  37. EXPECT_FLOAT_EQ(res1, 0.0);
  38. EXPECT_FLOAT_EQ(res2, 0.0);
  39. EXPECT_FLOAT_EQ(res3, 0.0);
  40. EXPECT_FLOAT_EQ(res4, 2.0);
  41. /* deinit */
  42. obj_deinit(pikaMain);
  43. EXPECT_EQ(pikaMemNow(), 0);
  44. }
  45. TEST(PikaMath, quat_sub) {
  46. /* init */
  47. g_PikaMemInfo.heapUsedMax = 0;
  48. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  49. /* run */
  50. obj_run(pikaMain,
  51. "import PikaMath\n"
  52. "a = PikaMath.Quaternion()\n"
  53. "b = PikaMath.Quaternion()\n"
  54. "a.sub(b)\n"
  55. "res1=a.get(0)\n"
  56. "res2=a.get(1)\n"
  57. "res3=a.get(2)\n"
  58. "res4=a.get(3)\n");
  59. /* collect */
  60. float res1 = obj_getFloat(pikaMain, "res1");
  61. float res2 = obj_getFloat(pikaMain, "res2");
  62. float res3 = obj_getFloat(pikaMain, "res3");
  63. float res4 = obj_getFloat(pikaMain, "res4");
  64. /* assert */
  65. EXPECT_FLOAT_EQ(res1, 0.0);
  66. EXPECT_FLOAT_EQ(res2, 0.0);
  67. EXPECT_FLOAT_EQ(res3, 0.0);
  68. EXPECT_FLOAT_EQ(res4, 0.0);
  69. /* deinit */
  70. obj_deinit(pikaMain);
  71. EXPECT_EQ(pikaMemNow(), 0);
  72. }
  73. TEST(PikaMath, quat_mul) {
  74. /* init */
  75. g_PikaMemInfo.heapUsedMax = 0;
  76. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  77. /* run */
  78. obj_run(pikaMain,
  79. "import PikaMath\n"
  80. "a = PikaMath.Quaternion()\n"
  81. "b = PikaMath.Quaternion()\n"
  82. "a.mul(b)\n"
  83. "res1=a.get(0)\n"
  84. "res2=a.get(1)\n"
  85. "res3=a.get(2)\n"
  86. "res4=a.get(3)\n");
  87. /* collect */
  88. float res1 = obj_getFloat(pikaMain, "res1");
  89. float res2 = obj_getFloat(pikaMain, "res2");
  90. float res3 = obj_getFloat(pikaMain, "res3");
  91. float res4 = obj_getFloat(pikaMain, "res4");
  92. /* assert */
  93. EXPECT_FLOAT_EQ(res1, 0.0);
  94. EXPECT_FLOAT_EQ(res2, 0.0);
  95. EXPECT_FLOAT_EQ(res3, 0.0);
  96. EXPECT_FLOAT_EQ(res4, 1.0);
  97. /* deinit */
  98. obj_deinit(pikaMain);
  99. EXPECT_EQ(pikaMemNow(), 0);
  100. }
  101. TEST(PikaMath, quat_set) {
  102. /* init */
  103. g_PikaMemInfo.heapUsedMax = 0;
  104. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  105. /* run */
  106. obj_run(pikaMain,
  107. "import PikaMath\n"
  108. "a = PikaMath.Quaternion()\n"
  109. "a.set(1,1,1,1)\n"
  110. "res1=a.get(0)\n"
  111. "res2=a.get(1)\n"
  112. "res3=a.get(2)\n"
  113. "res4=a.get(3)\n");
  114. /* collect */
  115. float res1 = obj_getFloat(pikaMain, "res1");
  116. float res2 = obj_getFloat(pikaMain, "res2");
  117. float res3 = obj_getFloat(pikaMain, "res3");
  118. float res4 = obj_getFloat(pikaMain, "res4");
  119. /* assert */
  120. EXPECT_FLOAT_EQ(res1, 1);
  121. EXPECT_FLOAT_EQ(res2, 1);
  122. EXPECT_FLOAT_EQ(res3, 1);
  123. EXPECT_FLOAT_EQ(res4, 1);
  124. /* deinit */
  125. obj_deinit(pikaMain);
  126. EXPECT_EQ(pikaMemNow(), 0);
  127. }
  128. TEST(PikaMath, quat_dot) {
  129. /* init */
  130. g_PikaMemInfo.heapUsedMax = 0;
  131. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  132. /* run */
  133. obj_run(pikaMain,
  134. "import PikaMath\n"
  135. "a = PikaMath.Quaternion()\n"
  136. "b = PikaMath.Quaternion()\n"
  137. "a.set(1,2,3,4)\n"
  138. "b.set(5,6,7,8)\n"
  139. "res1=a.dot(b)\n");
  140. /* collect */
  141. float res1 = obj_getFloat(pikaMain, "res1");
  142. /* assert */
  143. EXPECT_FLOAT_EQ(res1, 70.0);
  144. /* deinit */
  145. obj_deinit(pikaMain);
  146. EXPECT_EQ(pikaMemNow(), 0);
  147. }
  148. TEST(PikaMath, quat_inv) {
  149. /* init */
  150. g_PikaMemInfo.heapUsedMax = 0;
  151. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  152. /* run */
  153. obj_run(pikaMain,
  154. "import PikaMath\n"
  155. "a = PikaMath.Quaternion()\n"
  156. "a.inverse()\n"
  157. "res1=a.get(0)\n"
  158. "res2=a.get(1)\n"
  159. "res3=a.get(2)\n"
  160. "res4=a.get(3)\n");
  161. /* collect */
  162. float res1 = obj_getFloat(pikaMain, "res1");
  163. float res2 = obj_getFloat(pikaMain, "res2");
  164. float res3 = obj_getFloat(pikaMain, "res3");
  165. float res4 = obj_getFloat(pikaMain, "res4");
  166. /* assert */
  167. EXPECT_FLOAT_EQ(res1, 0.0);
  168. EXPECT_FLOAT_EQ(res2, 0.0);
  169. EXPECT_FLOAT_EQ(res3, 0.0);
  170. EXPECT_FLOAT_EQ(res4, 1.0);
  171. /* deinit */
  172. obj_deinit(pikaMain);
  173. EXPECT_EQ(pikaMemNow(), 0);
  174. }
  175. TEST(PikaMath, quat_cross) {
  176. /* init */
  177. g_PikaMemInfo.heapUsedMax = 0;
  178. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  179. /* run */
  180. obj_run(pikaMain,
  181. "import PikaMath\n"
  182. "a = PikaMath.Quaternion()\n"
  183. "b = PikaMath.Quaternion()\n"
  184. "a.set(2,3,4,1)\n"
  185. "b.set(6,7,8,5)\n"
  186. "a.crossproduct(b)\n"
  187. "res1=a.get(0)\n"
  188. "res2=a.get(1)\n"
  189. "res3=a.get(2)\n"
  190. "res4=a.get(3)\n");
  191. /* collect */
  192. float res1 = obj_getFloat(pikaMain, "res1");
  193. float res2 = obj_getFloat(pikaMain, "res2");
  194. float res3 = obj_getFloat(pikaMain, "res3");
  195. float res4 = obj_getFloat(pikaMain, "res4");
  196. /* assert */
  197. EXPECT_FLOAT_EQ(res1, -4.0);
  198. EXPECT_FLOAT_EQ(res2, 8.0);
  199. EXPECT_FLOAT_EQ(res3, -4.0);
  200. EXPECT_FLOAT_EQ(res4, 0.0);
  201. /* deinit */
  202. obj_deinit(pikaMain);
  203. EXPECT_EQ(pikaMemNow(), 0);
  204. }
  205. TEST(PikaMath, quat_toEuler) {
  206. /* init */
  207. g_PikaMemInfo.heapUsedMax = 0;
  208. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  209. /* run */
  210. obj_run(pikaMain,
  211. "import PikaMath\n"
  212. "a = PikaMath.Quaternion()\n"
  213. "a.set(0,0,1,0)\n"
  214. "res=a.toEuler()\n"
  215. "res1=res[0]\n"
  216. "res2=res[1]\n"
  217. "res3=res[2]\n");
  218. /* collect */
  219. float res1 = obj_getFloat(pikaMain, "res1");
  220. float res2 = obj_getFloat(pikaMain, "res2");
  221. float res3 = obj_getFloat(pikaMain, "res3");
  222. /* assert */
  223. EXPECT_FLOAT_EQ(res1, 0.0);
  224. EXPECT_FLOAT_EQ(res2, 0.0);
  225. EXPECT_FLOAT_EQ(res3, 3.1415927);
  226. /* deinit */
  227. obj_deinit(pikaMain);
  228. EXPECT_EQ(pikaMemNow(), 0);
  229. }
  230. TEST(PikaMath, quat_fromEuler) {
  231. /* init */
  232. g_PikaMemInfo.heapUsedMax = 0;
  233. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  234. /* run */
  235. obj_run(pikaMain,
  236. "import PikaMath\n"
  237. "a = PikaMath.Quaternion()\n"
  238. "a.fromEuler(0,0.0,0.0,3.1415927)\n"
  239. "res1=a.get(0)\n"
  240. "res2=a.get(1)\n"
  241. "res3=a.get(2)\n"
  242. "res4=a.get(3)\n");
  243. /* collect */
  244. float res1 = obj_getFloat(pikaMain, "res1");
  245. float res2 = obj_getFloat(pikaMain, "res2");
  246. float res3 = obj_getFloat(pikaMain, "res3");
  247. float res4 = obj_getFloat(pikaMain, "res4");
  248. /* assert */
  249. EXPECT_FLOAT_EQ(res1, 0.0);
  250. EXPECT_FLOAT_EQ(res2, 0.0);
  251. EXPECT_FLOAT_EQ(res3, 0.0);
  252. EXPECT_FLOAT_EQ(res4, 1.0);
  253. /* deinit */
  254. obj_deinit(pikaMain);
  255. EXPECT_EQ(pikaMemNow(), 0);
  256. }
  257. TEST_RUN_SINGLE_FILE_PASS(PikaMath,
  258. modbus,
  259. "test/python/PikaMath/modbus_convert.py")
  260. #endif
  261. TEST_END