def test_load_pyrnn_no_seqrecognizer(self): """ Test correct handling of non-SeqRecognizer pickles. """ pickle.dump(u'Iámnõtãrécðçnízer', self.temp) self.temp.close() models.load_pyrnn(self.temp.name)
def test_load_pyrnn_uncompressed(self): """ Test correct handling of uncompressed models. """ rnn = models.load_pyrnn(os.path.join(resources, 'model.pyrnn')) self.assertIsInstance(rnn, kraken.lib.lstm.SeqRecognizer)
def test_load_pyrnn_invalid(self): """ Test correct handling of non-pickle files. """ self.temp.write('adfhewf') models.load_pyrnn(self.temp.name)