test_fish.py

import unittest

from src.fish import solution
class MyTestCase(unittest.TestCase):
    def test_something(self):
        A = [4,3,2,1,5]
        B = [0,1,0,0,0]

        res = solution(A,B)

        self.assertEqual(2,res , res )

    def test_wrong1(self):
        A =  [8, 6, 5, 3, 2, 4, 7]
        B =  [1, 1, 1, 1, 1, 0, 0]
        self.assertEqual(1, solution(A, B))


    def test_wrong2(self):
        A = [8, 6, 4, 2, 1, 3, 5]
        B = [1, 1, 1, 1, 1, 0, 0]
        self.assertEqual(2, solution(A, B))

    def test_wrong3(self):
        A =  [60784, 686946, 40022, 400519, 657473, 336999, 36846, 8629, 915034, 998496, 924250, 822568, 619077, 276547,
            241586, 673773, 30494, 74609, 18604, 810348, 983816, 474571, 222168, 696835, 810107, 549971, 142600, 281896,
            987134, 690593, 476768, 58824, 508926, 51353, 771919, 511351, 976304, 595971, 178049, 967459, 234972,
            363128, 90978, 112527, 28143, 663596, 905973, 818820, 946713, 388246, 363322, 404614, 344178, 6674, 661321,
            621452, 955892, 162267, 312541, 418840, 476262, 341790, 208793, 423207, 558190, 958749, 364193, 356345,
            636930, 666411, 135069, 268254, 601457, 52058, 459262, 440055, 145302, 167563, 113832, 770673, 481024,
            374754, 842404, 912042, 690115, 189232, 102613, 726328, 815698, 171214, 500012, 384581, 427243, 163524,
            518997, 748528, 695896, 274048, 10860, 430154]
        B =  [0, 1, 1, 0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 1, 0, 1, 0,
            1, 0, 0, 1, 0, 0, 1, 1, 0, 1, 1, 0, 0, 0, 0, 1, 0, 1, 0,
            0, 1, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0,
            1, 1, 1, 0, 1, 0, 1, 0, 0, 0, 1, 0, 1, 0, 1, 0, 0, 0, 1,
            0, 0, 0, 0, 0, 1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 1, 0, 0,
            0, 0, 1, 1, 1]
        self.assertEqual(16, solution(A, B))


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