Skip to content

Commit

Permalink
test: add tests for reading vignette parameters in ZemaxFileReader
Browse files Browse the repository at this point in the history
  • Loading branch information
HarrisonKramer committed Jan 22, 2025
1 parent 02ee335 commit 31bb583
Showing 1 changed file with 30 additions and 0 deletions.
30 changes: 30 additions & 0 deletions tests/test_fileio.py
Original file line number Diff line number Diff line change
Expand Up @@ -239,6 +239,36 @@ def test_read_surf_param(self, zemax_file_reader):
val = zemax_file_reader._current_surf_data['param_0']
assert val == 2

def test_read_vignette_decenter_x(self, zemax_file_reader):
data = ['VDXN', '1', '2']
zemax_file_reader._read_vignette_decenter_x(data)
val = zemax_file_reader.data['fields']['vignette_decenter_x']
assert val == [1, 2]

def test_read_vignette_decenter_y(self, zemax_file_reader):
data = ['VDYN', '1', '2']
zemax_file_reader._read_vignette_decenter_y(data)
val = zemax_file_reader.data['fields']['vignette_decenter_y']
assert val == [1, 2]

def test_read_vignette_compress_x(self, zemax_file_reader):
data = ['VCXN', '1', '2']
zemax_file_reader._read_vignette_compress_x(data)
val = zemax_file_reader.data['fields']['vignette_compress_x']
assert val == [1, 2]

def test_read_vignette_compress_y(self, zemax_file_reader):
data = ['VCYN', '1', '2']
zemax_file_reader._read_vignette_compress_y(data)
val = zemax_file_reader.data['fields']['vignette_compress_y']
assert val == [1, 2]

def test_read_vignette_tangent_angle(self, zemax_file_reader):
data = ['VANN', '1', '2']
zemax_file_reader._read_vignette_tangent_angle(data)
val = zemax_file_reader.data['fields']['vignette_tangent_angle']
assert val == [1, 2]


class TestZemaxToOpticConverter:
def test_generate_lens(self, zemax_file_reader):
Expand Down

0 comments on commit 31bb583

Please sign in to comment.