Skip to content

Commit c4ebb0d

Browse files
committed
editing robot test for raw img integration
1 parent 98e4d5c commit c4ebb0d

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

ledsa/core/led_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,8 @@ def read_file(filename, channel, colordepth=14):
4747
8bit is default range for JPG. Bayer array is a 2D array where
4848
all channel values except the selected channel are masked.
4949
"""
50-
extension = os.path.splitext(filename)[1]
51-
if extension in ['.JPG','.JPEG', '.jpg', '.jpeg']:
50+
extension = os.path.splitext(filename)[-1]
51+
if extension in ['.JPG','.JPEG', '.jpg', '.jpeg', '.PNG', '.png']:
5252
data = plt.imread(filename)
5353
elif extension in ['.CR2']:
5454
with rawpy.imread(filename) as raw:

ledsa/tests/AcceptanceTests/LedsaATestLibrary.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def create_test_image(self, amount=1):
2424

2525
@keyword
2626
def create_and_fill_config(self):
27-
conf = ConfigData(False, '.', 10, 1, False, 1, 'test_img_0.png', None, None, 0, None, 'test_img_{}.png',
27+
conf = ConfigData(False, '.', 10, 0.25, 1, False, 1, 'test_img_0.png', None, None, 0, None, 'test_img_{}.png',
2828
0, 0, 0, 0)
2929
conf.set('analyse_positions', ' line_edge_indices', '0 2')
3030
conf.set('DEFAULT', ' date', '2018:11:27')

0 commit comments

Comments
 (0)