55 lines
3.9 KiB
Python
55 lines
3.9 KiB
Python
|
import unittest
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
from urh.ainterpretation import AutoInterpretation
|
||
|
|
|
||
|
|
class TestAutoInterpretation(unittest.TestCase):
|
||
|
|
def __run_merge(self, data):
|
||
|
|
return list(AutoInterpretation.merge_plateau_lengths(np.array(data, dtype=np.uint64)))
|
||
|
|
|
||
|
|
def test_merge_plateau_lengths(self):
|
||
|
|
self.assertEqual(AutoInterpretation.merge_plateau_lengths([]), [])
|
||
|
|
self.assertEqual(AutoInterpretation.merge_plateau_lengths([42]), [42])
|
||
|
|
self.assertEqual(AutoInterpretation.merge_plateau_lengths([100, 100, 100]), [100, 100, 100])
|
||
|
|
self.assertEqual(self.__run_merge([100, 49, 1, 50, 100]), [100, 100, 100])
|
||
|
|
self.assertEqual(self.__run_merge([100, 48, 2, 50, 100]), [100, 100, 100])
|
||
|
|
self.assertEqual(self.__run_merge([100, 100, 67, 1, 10, 1, 21]), [100, 100, 100])
|
||
|
|
self.assertEqual(self.__run_merge([100, 100, 67, 1, 10, 1, 21, 100, 50, 1, 49]), [100, 100, 100, 100, 100])
|
||
|
|
|
||
|
|
def test_estimate_tolerance_from_plateau_lengths(self):
|
||
|
|
self.assertEqual(AutoInterpretation.estimate_tolerance_from_plateau_lengths([]), None)
|
||
|
|
self.assertEqual(AutoInterpretation.estimate_tolerance_from_plateau_lengths([10]), None)
|
||
|
|
self.assertEqual(AutoInterpretation.estimate_tolerance_from_plateau_lengths([100, 49, 1, 50, 100]), 1)
|
||
|
|
self.assertEqual(AutoInterpretation.estimate_tolerance_from_plateau_lengths([100, 49, 2, 50, 100]), 2)
|
||
|
|
self.assertEqual(AutoInterpretation.estimate_tolerance_from_plateau_lengths([100, 49, 2, 50, 100, 1]), 2)
|
||
|
|
self.assertEqual(AutoInterpretation.estimate_tolerance_from_plateau_lengths([8, 8, 6, 1, 1]), 1)
|
||
|
|
|
||
|
|
def test_tolerant_greatest_common_divisor(self):
|
||
|
|
self.assertEqual(AutoInterpretation.get_tolerant_greatest_common_divisor([]), 1)
|
||
|
|
self.assertEqual(AutoInterpretation.get_tolerant_greatest_common_divisor([22]), 1)
|
||
|
|
self.assertEqual(AutoInterpretation.get_tolerant_greatest_common_divisor([10, 5, 5]), 5)
|
||
|
|
self.assertEqual(AutoInterpretation.get_tolerant_greatest_common_divisor([100, 100, 100]), 100)
|
||
|
|
self.assertEqual(AutoInterpretation.get_tolerant_greatest_common_divisor([100, 100, 200, 300, 100, 400]), 100)
|
||
|
|
self.assertEqual(AutoInterpretation.get_tolerant_greatest_common_divisor([100, 101, 100, 100]), 100)
|
||
|
|
self.assertEqual(AutoInterpretation.get_tolerant_greatest_common_divisor([100, 101, 202, 301, 100, 500]), 100)
|
||
|
|
|
||
|
|
def test_get_bit_length_from_plateau_length(self):
|
||
|
|
self.assertEqual(AutoInterpretation.get_bit_length_from_plateau_lengths([]), 0)
|
||
|
|
self.assertEqual(AutoInterpretation.get_bit_length_from_plateau_lengths([42]), 42)
|
||
|
|
plateau_lengths = np.array([2, 1, 2, 73, 1, 26, 100, 40, 1, 59, 100, 47, 1, 52, 67, 1, 10, 1, 21, 33, 1, 66, 100, 5, 1, 3, 1, 48, 1, 27, 1, 8], dtype=np.uint64)
|
||
|
|
merged_lengths = AutoInterpretation.merge_plateau_lengths(plateau_lengths)
|
||
|
|
self.assertEqual(AutoInterpretation.get_bit_length_from_plateau_lengths(merged_lengths), 100)
|
||
|
|
|
||
|
|
|
||
|
|
plateau_lengths = np.array([1, 292, 331, 606, 647, 286, 645, 291, 334, 601, 339, 601, 338, 602, 337, 603, 338, 604, 336, 605, 337, 600, 338, 605, 646], dtype=np.uint64)
|
||
|
|
merged_lengths = AutoInterpretation.merge_plateau_lengths(plateau_lengths)
|
||
|
|
self.assertEqual(AutoInterpretation.get_bit_length_from_plateau_lengths(merged_lengths), 300)
|
||
|
|
|
||
|
|
plateau_lengths = np.array([3, 8, 8, 8, 8, 8, 8, 8, 8, 16, 8, 8, 16, 32, 8, 8, 8, 8, 8, 24, 8, 24, 8, 24, 8, 24, 8, 24, 16, 16, 24, 8], dtype=np.uint64)
|
||
|
|
merged_lengths = AutoInterpretation.merge_plateau_lengths(plateau_lengths)
|
||
|
|
self.assertEqual(AutoInterpretation.get_bit_length_from_plateau_lengths(merged_lengths), 8)
|
||
|
|
|
||
|
|
def test_get_bit_length_from_merged_plateau_lengths(self):
|
||
|
|
merged_lengths = np.array([40, 40, 40, 40, 40, 30, 50, 30, 90, 40, 40, 80, 160, 30, 50, 30], dtype=np.uint64)
|
||
|
|
self.assertEqual(AutoInterpretation.get_bit_length_from_plateau_lengths(merged_lengths), 40)
|