def test_invalid_max_util_drop(self):
''' Invalid max_util_drop. '''
with self.assertRaisesRegex(ValueError,
'InterLayerPipeline: .*max_util_drop.*'):
_ = InterLayerPipeline(self.net['net1'], self.batch_size,
self.resource, max_util_drop=1.1)
with self.assertRaisesRegex(ValueError,
'InterLayerPipeline: .*max_util_drop.*'):
_ = InterLayerPipeline(self.net['net1'], self.batch_size,
self.resource, max_util_drop=-0.1)
def test_valid_args(self):
''' Valid arguments. '''
ilp = InterLayerPipeline(self.net['net1'], self.batch_size,
self.resource, max_util_drop=0.1)
self.assertIs(ilp.network, self.net['net1'])
self.assertEqual(ilp.batch_size, self.batch_size)
self.assertIs(ilp.resource, self.resource)
self.assertEqual(ilp.max_util_drop, 0.1)
def test_invalid_resource(self):
''' Invalid resource. '''
with self.assertRaisesRegex(TypeError,
'InterLayerPipeline: .*resource.*'):
_ = InterLayerPipeline(self.net['net1'], self.batch_size,
PhyDim2(1, 1))
def test_invalid_network(self):
''' Invalid network. '''
with self.assertRaisesRegex(TypeError,
'InterLayerPipeline: .*network.*'):
_ = InterLayerPipeline(self.net['net1'].input_layer(),
self.batch_size, self.resource)
def _make_ilp(self, network):
''' Make an InterLayerPipeline instance. '''
return InterLayerPipeline(network, self.batch_size, self.resource)