def test_parameter_space_data_shape(self): options = self.framework_options_setup() plugin = tu.plugin_runner_load_plugin(options) tu.plugin_setup(plugin) out_dataset = plugin.get_out_datasets()[0] self.assertEqual((160, 1, 160, 2, 3), out_dataset.get_shape())