def test_dictionary_all_docs(): d = Dictionary() assert_eq([], d.all_docs()) d.add_term('asdf', 1, 1) assert_eq([1], d.all_docs()) d.add_term('asdf', 2, 1) assert_eq([1, 2], d.all_docs()) d.add_term('qwer', 1, 1) d.add_term('zxcv', 1, 1) assert_eq([1, 2], d.all_docs())
def test_dictionary_to_json_from_json(): d = Dictionary() d.add_term('asdf', 1, 1) d.add_term('asdf', 2, 1) d.add_term('qwer', 1, 1) d.add_term('zxcv', 1, 1) d2 = Dictionary.from_json(d.to_json()) assert_eq(d2.all_docs(), d.all_docs()) assert_eq(d2.all_terms(), d.all_terms()) assert_eq(d2.get_frequency('asdf'), d.get_frequency('asdf')) assert_eq(d2.get_frequency('qwer'), d.get_frequency('qwer')) assert_eq(d2.get_frequency('zxcv'), d.get_frequency('zxcv')) assert_eq(d2.get_head('asdf'), d.get_head('asdf')) assert_eq(d2.get_head('qwer'), d.get_head('qwer')) assert_eq(d2.get_head('zxcv'), d.get_head('zxcv')) assert_eq(d2.get_tail('asdf'), d.get_tail('asdf')) assert_eq(d2.get_tail('qwer'), d.get_tail('qwer')) assert_eq(d2.get_tail('zxcv'), d.get_tail('zxcv'))