Exemplo n.º 1
0
    def test_seven_kernel_of_two_and_stride_three_and_padding_two(self):
        expected = ([None, None], [1, 2], [4, 5], [None, None])
        result = []

        for kernel_hover in convolved_1d(list(range(7)), kernel_size=2, padding=2, stride=3):
            result.append(kernel_hover)
        result = tuple(result)

        self.assertEqual(expected, result)
Exemplo n.º 2
0
    def test_seven_kernel_of_two(self):
        expected = tuple([a, b] for a, b in zip(list(range(0, 6)), list(range(1, 7))))
        result = []

        for kernel_hover in convolved_1d(list(range(7)), kernel_size=2, padding=0, stride=1):
            result.append(kernel_hover)
        result = tuple(result)

        self.assertEqual(expected, result)
Exemplo n.º 3
0
    def test_seven_for_with_stride_two(self):
        expected = tuple(range(0, 7, 2))
        result = []

        for kernel_hover in convolved_1d(list(range(7)), kernel_size=1, padding=0, stride=2):
            result.append(*kernel_hover)
        result = tuple(result)

        self.assertEqual(expected, result)
Exemplo n.º 4
0
    def test_seven_for_with_padding_one(self):
        expected = tuple([42] + list(range(0, 7)) + [42])
        result = []

        for kernel_hover in convolved_1d(list(range(7)), kernel_size=1, padding=1, stride=1, default_value=42):
            result.append(*kernel_hover)
        result = tuple(result)

        self.assertEqual(expected, result)
Exemplo n.º 5
0
    def test_trivial_loop(self):
        expected = tuple(range(7))
        result = []

        for kernel_hover in convolved_1d(expected, kernel_size=1, padding=0, stride=1):
            result.append(*kernel_hover)
        result = tuple(result)

        self.assertEqual(expected, result)