def test_gets(self): from wordvector import WordVector dictionary = { 'the': 0, 'quick': 1, 'brown': 2, 'fox': 3, 'jumped': 4, 'over': 5 } embed_matrix = np.array([[1.0, 1.01], [2.0, 2.0], [2.0, 2.1], [1.0, 0.0], [0, 1.01], [-1.0, 0.0]]) word_embedding = WordVector(embed_matrix, dictionary) d = word_embedding.get_dict() dr = word_embedding.get_reverse_dict() em = word_embedding.get_embed() d.pop('the') # mutate, check that copies were returned dr.pop(1) em[0, 0] = 10 d = word_embedding.get_dict() dr = word_embedding.get_reverse_dict() em = word_embedding.get_embed() self.assertEqual(6, len(d), 'wrong dictionary length') self.assertEqual(6, len(dr), 'wrong dictionary length') self.assertEqual(1.0, em[0, 0], 'wrong value in embed matrix') self.assertEqual(3, d['fox'], 'wrong value from dictionary') self.assertEqual('jumped', dr[4], 'wrong value from reverse dictionary')