Closes #2570: bigint_agg_test.py Conversion for new test framework

PROTO_tests/tests/bigint_agg_test.py

@@ -0,0 +1,61 @@
+import arkouda as ak
+import numpy as np
+
+import pytest
+
+
+class TestBigInt:
+    @pytest.mark.parametrize("size", pytest.prob_size)
+    def test_negative(self, size):
+        # test with negative bigint values
+        arr = -1 * ak.randint(0, 2**32, size)
+        bi_neg = ak.cast(arr, ak.bigint)
+        res = gather_scatter(bi_neg)
+        assert bi_neg.to_list() == res.to_list()
+
+    @pytest.mark.parametrize("size", pytest.prob_size)
+    def test_large(self, size):
+        # test with 256 bit bigint values
+        top_bits = ak.randint(0, 2**32, size, dtype=ak.uint64)
+        mid_bits1 = ak.randint(0, 2**32, size, dtype=ak.uint64)
+        mid_bits2 = ak.randint(0, 2**32, size, dtype=ak.uint64)
+        bot_bits = ak.randint(0, 2**32, size, dtype=ak.uint64)
+        bi_arr = ak.bigint_from_uint_arrays([top_bits, mid_bits1, mid_bits2, bot_bits])
+        res = gather_scatter(bi_arr)
+        assert bi_arr.to_list() == res.to_list()
+
+    @pytest.mark.parametrize("size", pytest.prob_size)
+    def test_zero(self, size):
+        # test all zero bigint assignments
+        all_zero = ak.zeros(size, dtype=ak.bigint)
+        res = gather_scatter(all_zero)
+        assert all_zero.to_list() == res.to_list()
+
+    def test_variable_sized(self):
+        # 5 bigints of differing number of limbs
+        bits1 = ak.array([0, 0, 0, 0, 1], dtype=ak.uint64)
+        bits2 = ak.array([0, 0, 0, 1, 1], dtype=ak.uint64)
+        bits3 = ak.array([0, 0, 1, 1, 1], dtype=ak.uint64)
+        bits4 = ak.array([0, 1, 1, 1, 1], dtype=ak.uint64)
+        bits5 = ak.array([1, 1, 1, 1, 1], dtype=ak.uint64)
+        bi_arr = ak.bigint_from_uint_arrays([bits1, bits2, bits3, bits4, bits5])
+        res = gather_scatter(bi_arr)
+        assert bi_arr.to_list() == res.to_list()
+
+    @pytest.mark.parametrize("size", pytest.prob_size)
+    def test_change_size(self, size):
+        # Go from 256 bigint values down to just 1
+        bits = ak.randint(0, 2**32, size, dtype=ak.uint64)
+        bi_arr = ak.bigint_from_uint_arrays([bits, bits, bits, bits])
+        res = ak.ones_like(bi_arr)
+        bi_arr[:] = res
+        assert bi_arr.to_list() == res.to_list()
+
+
+def gather_scatter(a):
+    rev = ak.array(np.arange(len(a) - 1, -1, -1))
+    a2 = a[rev]
+    res = ak.zeros(len(a), dtype=a.dtype)
+    res[:] = a2
+    res[rev] = a2
+    return res