class RawDatasetReaderPartialTest(unittest.TestCase):
def setUp(self):
dataset_filepath = SurealConfig.test_resource_path(
'NFLX_dataset_public_raw_PARTIAL.py')
self.dataset = import_python_file(dataset_filepath)
self.dataset_reader = RawDatasetReader(self.dataset)
def test_read_dataset_stats(self):
self.assertEqual(self.dataset_reader.num_ref_videos, 7)
self.assertEqual(self.dataset_reader.max_content_id_of_ref_videos, 8)
self.assertEqual(self.dataset_reader.num_dis_videos, 51)
self.assertEqual(self.dataset_reader.num_observers, 26)
def test_opinion_score_2darray(self):
os_2darray = self.dataset_reader.opinion_score_2darray
self.assertAlmostEqual(np.mean(os_2darray),
3.4871794871794872,
places=4)
self.assertAlmostEqual(np.mean(np.std(os_2darray, axis=1)),
0.65626252041788125,
places=4)
def test_dis_videos_content_ids(self):
content_ids = self.dataset_reader.content_id_of_dis_videos
self.assertAlmostEqual(np.mean(content_ids),
3.9215686274509802,
places=4)
def test_disvideo_is_refvideo(self):
l = self.dataset_reader.disvideo_is_refvideo
self.assertTrue(all(l[0:7]))
def test_ref_score(self):
self.assertEqual(self.dataset_reader.ref_score, 5.0)
def test_to_persubject_dataset_wrong_dim(self):
with self.assertRaises(AssertionError):
dataset = self.dataset_reader.to_persubject_dataset(np.zeros(3000))
self.assertEqual(len(dataset.dis_videos), 2054)
def test_to_persubject_dataset(self):
dataset = self.dataset_reader.to_persubject_dataset(np.zeros([79, 26]))
self.assertEqual(len(dataset.dis_videos), 1326)
class RawDatasetReaderTest(unittest.TestCase):
def setUp(self):
dataset_filepath = SurealConfig.test_resource_path(
'NFLX_dataset_public_raw.py')
self.dataset = import_python_file(dataset_filepath)
self.dataset_reader = RawDatasetReader(self.dataset)
def test_read_dataset_stats(self):
self.assertEqual(self.dataset_reader.num_ref_videos, 9)
self.assertEqual(self.dataset_reader.max_content_id_of_ref_videos, 8)
self.assertEqual(self.dataset_reader.num_dis_videos, 79)
self.assertEqual(self.dataset_reader.num_observers, 26)
def test_opinion_score_2darray(self):
os_2darray = self.dataset_reader.opinion_score_2darray
self.assertAlmostEqual(np.mean(os_2darray),
3.544790652385589,
places=4)
self.assertAlmostEqual(np.mean(np.std(os_2darray, axis=1)),
0.64933186478291516,
places=4)
def test_dis_videos_content_ids(self):
content_ids = self.dataset_reader.content_id_of_dis_videos
self.assertAlmostEqual(np.mean(content_ids),
3.8607594936708862,
places=4)
def test_disvideo_is_refvideo(self):
l = self.dataset_reader.disvideo_is_refvideo
self.assertTrue(all(l[0:9]))
def test_ref_score(self):
self.assertEqual(self.dataset_reader.ref_score, 5.0)
def test_to_persubject_dataset_wrong_dim(self):
with self.assertRaises(AssertionError):
dataset = self.dataset_reader.to_persubject_dataset(np.zeros(3000))
self.assertEqual(len(dataset.dis_videos), 2054)
def test_to_persubject_dataset(self):
dataset = self.dataset_reader.to_persubject_dataset(np.zeros([79, 26]))
self.assertEqual(len(dataset.dis_videos), 2054)