(self)
| 406 | self.assertEqual(set(current_ids), set(selected_items)) |
| 407 | |
| 408 | def test_pickle(self): |
| 409 | user_items = get_checker_board(50) |
| 410 | model = self._get_model() |
| 411 | model.fit(user_items, show_progress=False) |
| 412 | |
| 413 | pickled = pickle.dumps(model) |
| 414 | reloaded = pickle.loads(pickled) |
| 415 | |
| 416 | # make sure we can call methods on the reloaded index, and get the same results back |
| 417 | # (https://github.com/benfred/implicit/issues/631) |
| 418 | ids, _ = model.recommend(0, user_items[0]) |
| 419 | reloaded_ids, _ = reloaded.recommend(0, user_items[0]) |
| 420 | assert_array_equal(ids, reloaded_ids) |
| 421 | |
| 422 | ids, _ = model.similar_items(0) |
| 423 | reloaded_ids, _ = reloaded.similar_items(0) |
| 424 | assert_array_equal(ids, reloaded_ids) |
| 425 | |
| 426 | def test_pickle_unfitted_model(self): |
| 427 | model = self._get_model() |
nothing calls this directly
no test coverage detected