Skip to content

Commit 503ea57

Browse files
authored
Update test_sequences.py
1 parent a67ac1d commit 503ea57

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

deeptrack/tests/test_sequences.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,10 @@ def test_Sequence(self):
3030
def get_rotation(sequence_length, previous_value):
3131
return previous_value + 2 * np.pi / sequence_length
3232

33-
rotating_ellipse = sequences.Sequential(ellipse, rotation=get_rotation)
33+
rotating_ellipse = ellipse.to_sequential(rotation=get_rotation)
34+
3435
imaged_rotating_ellipse = optics(rotating_ellipse)
36+
3537
imaged_rotating_ellipse_sequence = sequences.Sequence(
3638
imaged_rotating_ellipse, sequence_length=5
3739
)
@@ -64,9 +66,7 @@ def get_rotation(sequence_length, previous_value):
6466
def get_intensity(rotation):
6567
return rotation * 2
6668

67-
rotating_ellipse = sequences.Sequential(
68-
ellipse, rotation=get_rotation, intensity=get_intensity
69-
)
69+
rotating_ellipse = ellipse.to_sequential(rotation=get_rotation, intensity=get_intensity)
7070
imaged_rotating_ellipse = optics(rotating_ellipse)
7171
imaged_rotating_ellipse_sequence = sequences.Sequence(
7272
imaged_rotating_ellipse, sequence_length=5
@@ -100,9 +100,7 @@ def get_rotation(sequence_length, previous_value):
100100
def get_intensity(rotation):
101101
return rotation * 2
102102

103-
rotating_ellipse = sequences.Sequential(
104-
ellipse, rotation=get_rotation, intensity=get_intensity
105-
)
103+
rotating_ellipse = ellipse.to_sequential(rotation=get_rotation, intensity=get_intensity)
106104
imaged_rotating_ellipse = optics(rotating_ellipse ^ 2)
107105
imaged_rotating_ellipse_sequence = sequences.Sequence(
108106
imaged_rotating_ellipse, sequence_length=5
@@ -148,9 +146,8 @@ def get_rotation(sequence_length, previous_value):
148146
def get_intensity(rotation):
149147
return rotation * 2
150148

151-
rotating_ellipse = sequences.Sequential(
152-
ellipse, rotation=get_rotation, intensity=get_intensity
153-
)
149+
rotating_ellipse = ellipse.to_sequential(rotation=get_rotation, intensity=get_intensity)
150+
154151
imaged_rotating_ellipse = optics(rotating_ellipse ^ 2)
155152
imaged_rotating_ellipse_sequence = sequences.Sequence(
156153
imaged_rotating_ellipse, sequence_length=5

0 commit comments

Comments
 (0)