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)