def test_cur_grid_has_layers(self):
prj = ChartProject(output_folder=testing.output_data_folder())
with self.assertRaises(RuntimeError):
prj.cur_grid_has_depth_layer()
with self.assertRaises(RuntimeError):
prj.cur_grid_has_product_uncertainty_layer()
with self.assertRaises(RuntimeError):
prj.cur_grid_has_density_layer()
with self.assertRaises(RuntimeError):
prj.cur_grid_has_tvu_qc_layer()
with self.assertRaises(RuntimeError):
prj.cur_grid_tvu_qc_layers()
with self.assertRaises(RuntimeError):
prj.set_cur_grid_tvu_qc_name("fake")
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
prj.set_cur_grid(testing.input_test_files(".bag")[0])
prj.open_to_read_cur_grid()
self.assertTrue(prj.cur_grid_has_depth_layer())
self.assertTrue(prj.cur_grid_has_product_uncertainty_layer())
self.assertFalse(prj.cur_grid_has_density_layer())
self.assertFalse(prj.cur_grid_has_tvu_qc_layer())
self.assertEqual(len(prj.cur_grid_tvu_qc_layers()), 0)
self.assertTrue(prj.has_bag_grid())
self.assertFalse(prj.has_csar_grid())
def test_cur_grid_shape(self):
prj = ChartProject(output_folder=testing.output_data_folder())
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
prj.set_cur_grid(testing.input_test_files(".bag")[0])
prj.open_to_read_cur_grid()
self.assertEquals(prj.cur_grid_shape[0], 0)
self.assertEquals(prj.cur_grid_shape[1], 0)
def test_grid_list_add_real(self):
prj = ChartProject(output_folder=testing.output_data_folder())
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
self.assertEqual(len(prj.grid_list), 1)
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
self.assertEqual(len(prj.grid_list), 1)
prj.remove_from_grid_list(testing.input_test_files(".bag")[0])
self.assertEqual(len(prj.grid_list), 0)
def test_select_layers_in_cur_grid(self):
prj = ChartProject(output_folder=testing.output_data_folder())
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
prj.set_cur_grid(testing.input_test_files(".bag")[0])
prj.open_to_read_cur_grid()
self.assertEqual(len(prj.selected_layers_in_cur_grid), 0)
prj.selected_layers_in_cur_grid = [
layer_types['depth'],
]
self.assertEqual(len(prj.selected_layers_in_cur_grid), 1)
def test_grid_read(self):
prj = ChartProject(output_folder=testing.output_data_folder())
self.assertTrue(prj.cur_grid is None)
self.assertFalse(prj.has_grid())
with self.assertRaises(RuntimeError):
_ = prj.cur_grid_basename
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
prj.set_cur_grid(testing.input_test_files(".bag")[0])
prj.open_to_read_cur_grid()
self.assertTrue(prj.cur_grid is not None)
self.assertTrue(prj.has_grid())
self.assertGreater(len(prj.cur_grid_basename), 0)
from hyo2.qc.common import default_logging
import logging
default_logging.load()
logger = logging.getLogger()
from hyo2.qc.chart.project import ChartProject
from hyo2.qc.common import testing
prj = ChartProject(output_folder=testing.output_data_folder())
input_bag_file = testing.input_test_files('.bag')[0]
logger.debug('input: %s' % input_bag_file)
prj.add_to_grid_list(input_bag_file)
prj.grid_truncate(version=2, decimal_places=0)
logger.debug(prj)
def test_grid_list_clear(self):
prj = ChartProject(output_folder=testing.output_data_folder())
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
prj.clear_grid_list()
self.assertEqual(len(prj.grid_list), 0)
def test_grid_list_add_fake(self):
prj = ChartProject(output_folder=testing.output_data_folder())
with self.assertRaises(RuntimeError):
prj.add_to_grid_list("fake/fake.bag")
def test_truncate_with_grids(self):
prj = ChartProject(output_folder=testing.output_data_folder())
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
self.assertTrue(prj.grid_truncate())
def test_truncate_with_wrong_version(self):
prj = ChartProject(output_folder=testing.output_data_folder())
prj.add_to_grid_list(testing.input_test_files(".bag")[0])
self.assertFalse(prj.grid_truncate(version=1))
