def test_handle_mxnet_error_index_fail(self): # Assert original error is raised when a ValueError is raised because string indexing fails with open('tests/data/exceptions/model_architecture_error_fail.txt', 'r') as tf: string = tf.read() error = CustomError(string) with self.assertRaises(CustomError): exceptions._handle_mxnet_error(error)
def test_handle_mxnet_error_random_string(self): # Assert original error is raised when string does not match expected error string because error not related to # weight shape mismatch string = 'cvbcbcbvb' error = CustomError(string) with self.assertRaises(CustomError): exceptions._handle_mxnet_error(error)
def test_handle_mxnet_error(self): # Assert ModelArchitectureError is raised with open('tests/data/exceptions/model_architecture_error.txt', 'r') as tf: string = tf.read() error = CustomError(string) with self.assertRaises(exceptions.ModelArchitectureError): exceptions._handle_mxnet_error(error)