cnn_test.py

""" Test module for Convolution Operation
"""
import unittest

from numpy.lib.type_check import imag
import cnn
import numpy as np

__author__ = "Alex Punnen"
__status__ = "Test"
class TestConv2D(unittest.TestCase):

    def test_conv2d(self):
        testConv2D = cnn.Conv2D()
        filter_size = 2       
        weight1 =  np.ones((filter_size,filter_size,3)) 
        image = np.arange(16).reshape(4, 4)
        print("Initial Image=",image)
        image = np.stack([image,image,image], axis=2) # to mimic RGB channel
        print("Image Shape=",image.shape)
        print("Image [0,0,0]=",image[0,0,0],"[1,0,1]=",image[1,0,1],"[2,0,2]=",image[2,0,2])
        print("Image [0,0,:]=",image[0,0,:])
        conv_activation= testConv2D.conv2d(image,weight1)
        print("Final Activation Shape=",conv_activation.shape)
        print("Final Activation =",conv_activation)




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