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_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)
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)
