class MissingDatasetReaderTest(unittest.TestCase): def setUp(self): dataset_filepath = config.ROOT + '/python/test/resource/NFLX_dataset_public_raw.py' dataset = import_python_file(dataset_filepath) np.random.seed(0) info_dict = { 'missing_probability': 0.1, } self.dataset_reader = MissingDataRawDatasetReader(dataset, input_dict=info_dict) def test_opinion_score_2darray(self): os_2darray = self.dataset_reader.opinion_score_2darray self.assertTrue(np.isnan(np.mean(os_2darray))) self.assertEquals(np.isnan(os_2darray).sum(), 201) def test_to_dataset(self): dataset = self.dataset_reader.to_dataset() old_scores = [dis_video['os'] for dis_video in self.dataset_reader.dataset.dis_videos] new_scores = [dis_video['os'] for dis_video in dataset.dis_videos] self.assertNotEquals(old_scores, new_scores)