def test_properties(self): reader = ImageReader(['image']) reader.initialise(SINGLE_MOD_DATA, SINGLE_MOD_TASK, single_mod_list) self.assertEqual(len(reader.output_list), 4) self.assertDictEqual(reader.shapes, {'image': (256, 168, 256, 1, 1)}) self.assertDictEqual(reader.tf_dtypes, {'image': tf.float32}) self.assertEqual(reader.names, ['image']) self.assertDictEqual(reader.input_sources, {'image': ('lesion', )}) self.assertEqual(reader.get_subject_id(1)[:4], 'Fin_')
def test_properties(self): reader = ImageReader(['image']) reader.initialise(SINGLE_MOD_DATA, SINGLE_MOD_TASK, single_mod_list) self.assertEqual(len(reader.output_list), 4) self.assertDictEqual(reader.shapes, {'image': (256, 168, 256, 1, 1)}) self.assertDictEqual(reader.tf_dtypes, {'image': tf.float32}) self.assertEqual(reader.names, ['image']) self.assertDictEqual(reader.input_sources, {'image': ('lesion',)}) self.assertEqual(reader.get_subject_id(1)[:4], 'Fin_')
def test_existing_csv(self): reader_for_csv = ImageReader(['image']) reader_for_csv.initialise_reader(SINGLE_MOD_DATA, SINGLE_MOD_TASK) reader = ImageReader(['image']) reader.initialise_reader(EXISTING_DATA, SINGLE_MOD_TASK) self.assertEquals(len(reader.output_list), 4) self.assertDictEqual(reader.shapes, {'image': (256, 168, 256, 1, 1)}) self.assertDictEqual(reader.tf_dtypes, {'image': tf.float32}) self.assertEqual(reader.names, ['image']) self.assertDictEqual(reader.input_sources, {'image': ('lesion', )}) self.assertEqual(reader.get_subject_id(1)[:4], 'Fin_')
def test_existing_csv(self): reader_for_csv = ImageReader(['image']) reader_for_csv.initialise(SINGLE_MOD_DATA, SINGLE_MOD_TASK, single_mod_list) reader = ImageReader(['image']) reader.initialise(EXISTING_DATA, SINGLE_MOD_TASK, existing_list) self.assertEqual(len(reader.output_list), 4) self.assertDictEqual(reader.spatial_ranks, {'image': 3}) self.assertDictEqual(reader.shapes, {'image': (256, 168, 256, 1, 1)}) self.assertDictEqual(reader.tf_dtypes, {'image': tf.float32}) self.assertEqual(reader.names, ('image', )) self.assertDictEqual(reader.input_sources, {'image': ('lesion', )}) self.assertEqual(reader.get_subject_id(1)[:4], 'Fin_') self.assertTrue(isinstance(reader.get_subject(1), dict))
def test_existing_csv(self): reader_for_csv = ImageReader(['image']) reader_for_csv.initialise( SINGLE_MOD_DATA, SINGLE_MOD_TASK, single_mod_list) reader = ImageReader(['image']) reader.initialise(EXISTING_DATA, SINGLE_MOD_TASK, existing_list) self.assertEqual(len(reader.output_list), 4) self.assertDictEqual(reader.spatial_ranks, {'image': 3}) self.assertDictEqual(reader.shapes, {'image': (256, 168, 256, 1, 1)}) self.assertDictEqual(reader.tf_dtypes, {'image': tf.float32}) self.assertEqual(reader.names, ('image',)) self.assertDictEqual(reader.input_sources, {'image': ('lesion',)}) self.assertEqual(reader.get_subject_id(1)[:4], 'Fin_') self.assertTrue(isinstance(reader.get_subject(1), dict))