def test_set_y_number_11():
# Create a Shape object from the header
shape = Shape({
'x_number': 143,
'x_start': 84000,
'x_step': 500,
'x_stop': 155000,
'y_number': 143,
'y_start': 455000,
'y_step': 500,
'y_stop': 526000
})
# Set the y number to 11
shape.set_y_number(11)
# Extract the dict from the Shape object
shape_dict = shape.to_dict()
# todo: Compare with the desired output
assert shape_dict == {
'x_number': 143,
'x_start': 84000,
'x_step': 500,
'x_stop': 155000,
'y_number': 11,
'y_start': 455000,
'y_step': (526000 - 455000) / 10,
'y_stop': 526000
}
def test_set_y_number_2():
# Create a Shape object from the header
shape = Shape({
'x_number': 143,
'x_start': 84000,
'x_step': 500,
'x_stop': 155000,
'y_number': 143,
'y_start': 455000,
'y_step': 500,
'y_stop': 526000
})
# Set the y number to 2
shape.set_y_number(2)
# Extract the dict from the Shape object
shape_dict = shape.to_dict()
assert shape_dict == {
'x_number': 143,
'x_start': 84000,
'x_step': 500,
'x_stop': 155000,
'y_number': 2,
'y_start': 455000,
'y_step': 526000 - 455000,
'y_stop': 526000
}
def test_set_y_number_nan():
# Create a Shape object from the header
shape = Shape({
'x_number': 143,
'x_start': 84000,
'x_step': 500,
'x_stop': 155000,
'y_number': 143,
'y_start': 455000,
'y_step': 500,
'y_stop': 526000
})
# Set the y number to NaN
shape.set_y_number(np.nan)
assert False
