'''
Created on Nov 5, 2015

A module to mock caffe structures for testing

@author: kashefy
'''
from .proto.caffe_pb2 import TRAIN, TEST

class proto:
    class caffe_pb2:
        class Datum:
            def SerializeToString(self):
                return "mock_this"
            def ParseFromString(self, val):
                return # do nothing
            
class Net:
    def forward(self):
        return "mock this"
            
class io:
    @staticmethod
    def array_to_datum(s):
        return proto.caffe_pb2.Datum()

'''
Created on Nov 5, 2015

A module to mock caffe structures for testing

@author: kashefy
'''
from .proto.caffe_pb2 import TRAIN, TEST, Datum
            
class Net:
    def forward(self):
        return "mock this"
            
class io:
    @staticmethod
    def array_to_datum(s):
        return Datum()

remove explicit declaration of Datum class for mocking