changed assertions from is to ==
__init__ was missing and matmat.gemm was throwing errors while calling gemm