from arrayutilities import Arr
import unittest
import random
from collections.abc import Mapping

class TestArr(unittest.TestCase):
    def setUp(self):
        self.test_array = {'a': 1, 'b': 2, 'c': 3, 'd': 4, 'e': 5}

    def test_random_without_number(self):
        result = Arr.random(self.test_array)
        self.assertIn(result, self.test_array.values(), "Should return a random value from the array")

    def test_random_with_number(self):
        number = 3
        result = Arr.random(self.test_array, number=number)
        self.assertEqual(len(result), number, f"Should return {number} random elements from the array")

    def test_random_with_number_and_preserve_keys(self):
        number = 3
        result = Arr.random(self.test_array, number=number, preserve_keys=True)
        self.assertIsInstance(result, Mapping, "Should return a dictionary when preserve_keys is True")
        self.assertEqual(len(result), number, f"Should return {number} random elements with preserved keys")

    def test_random_with_zero_number(self):
        number = 0
        result = Arr.random(self.test_array, number=number)
        self.assertEqual(result, [], "Should return an empty list when number is 0")

    def test_random_with_large_number(self):
        number = len(self.test_array) + 1
        with self.assertRaises(ValueError):
            Arr.random(self.test_array, number=number)

if __name__ == '__main__':
    unittest.main()