def test_weight_datatag_accessors(self):
frame = TFFramework()
frame.load(RunpyLoader, 'test/test_models/simple_nnet.py')
assert frame.get_weights(
mode='training') is not None, 'No weights returned.'
assert frame.get_weights(
mode='inference') is not None, 'No weights returned.'
def test_get_timing(self):
frame = TFFramework()
frame.load(RunpyLoader, 'test/test_models/simple_nnet.py')
assert frame.get_timing(
mode='training') is not None, 'No timing returned.'
assert frame.get_timing(
mode='inference') is not None, 'No timing returned.'
def test_load(self):
loaders = [
RunpyLoader,
]
identifiers = [
'test.test_models.empty_model',
'test/test_models/empty_model.py',
'test/../test/test_models/empty_model.py',
'test.test_models.simple_nnet',
'test/test_models/simple_nnet.py',
'test/../test/test_models/simple_nnet.py',
]
for loader in loaders:
for identifier in identifiers:
frame = TFFramework()
print 'Loading:', identifier
frame.load(loader, identifier)
assert frame.model.is_dnnamo_model, 'Model isnt actually a Dnnamo model'
assert isinstance(
frame.model,
DnnamoModel), 'Model isnt actually a Dnnamo model'
def test_failed_load(self):
with self.assertRaises(ImportError):
frame = TFFramework()
frame.load(RunpyLoader, 'nonexistent_module')
