def test_from_file(): sensor_grid = SensorGrid.from_file('./tests/assets/test_points.pts') assert sensor_grid.identifier == 'test_points' assert sensor_grid[0].pos == (0, 0, 0) assert sensor_grid[0].dir == (0, 0, 1) assert len(sensor_grid) == 3 assert sensor_grid[1].to_radiance() == '0.2 0.3 0.4 0.5 0.6 0.7' assert sensor_grid[2].to_radiance() == '-10.0 -5.0 0.0 -50.0 -60.0 -70.0'
def test_split_single_grid(): """Test splitting a sensor grid.""" sensor_grid = SensorGrid.from_file( './tests/assets/grid/sensor_grid_split.pts') folder = './tests/assets/temp' info = sensor_grid.to_files(folder, 1, 'single_grid') assert len(info) == 1 assert info[0]['count'] == sensor_grid.count
def test_split_grid(): """Test splitting a sensor grid.""" sensor_grid = SensorGrid.from_file( './tests/assets/grid/sensor_grid_split.pts') folder = './tests/assets/temp' info = sensor_grid.to_files(folder, 6, 'test_sensor_grid') assert len(info) == 6 for i in range(6 - 1): assert info[i]['count'] == 4 assert info[-1]['count'] == 1
def test_split_grid_single(): input_grid = './tests/assets/grid/sensor_grid_single.pts' output_folder = './tests/assets/temp' runner = CliRunner() result = runner.invoke(split_grid, [input_grid, '100', '--folder', output_folder]) assert result.exit_code == 0 # check the file is created and named correctly pts_file = os.path.join(output_folder, 'sensor_grid_single_0000.pts') assert os.path.isfile(pts_file) grid = SensorGrid.from_file(pts_file) assert grid.count == 21
def test_merge_grid(): base_name = 'sensor_grid_merge' input_folder = './tests/assets/grid' output_folder = './tests/assets/temp' runner = CliRunner() result = runner.invoke( merge_grid, [input_folder, base_name, '--folder', output_folder]) assert result.exit_code == 0 # check the file is created pts_file = os.path.join(output_folder, base_name + '.pts') assert os.path.isfile(pts_file) grid = SensorGrid.from_file(pts_file) assert grid.count == 21
def test_split_grid(): input_grid = './tests/assets/grid/sensor_grid_split.pts' output_folder = './tests/assets/temp' runner = CliRunner() result = runner.invoke(split_grid, [input_grid, '5', '--folder', output_folder]) assert result.exit_code == 0 # check the file is created for count in range(4): pts_file = os.path.join(output_folder, 'sensor_grid_split_%04d.pts' % count) assert os.path.isfile(pts_file) grid = SensorGrid.from_file(pts_file) if count != 3: assert grid.count == 5 else: assert grid.count == 6