def test_find_data_segment_with_nested_list_pattern_and_multiple_variables():
data = "A = [3]; X = [[0, 1, 0], [0, 0, 1]];"
pattern = get_nested_list_pattern()
assert find_data_segment(data, "X", pattern) == "[[0, 1, 0], [0, 0, 1]]"
def test_find_data_segment_with_list_of_strings_pattern_with_multiple_variables(
):
data = 'A = ["b", "a"]; BB = ["z", "zz", "zzz"];'
pattern = get_list_of_strings_pattern()
assert find_data_segment(data, "BB", pattern) == '["z", "zz", "zzz"]'
def test_find_data_segment_with_nested_list_pattern():
data = "Z = [[1, 1, 0], [0, 0, 1]];"
pattern = get_nested_list_pattern()
assert find_data_segment(data, "Z", pattern) == "[[1, 1, 0], [0, 0, 1]]"
def test_find_data_segment_with_list_of_strings_pattern():
data = 'A = ["xyz", "zyx"];'
pattern = get_list_of_strings_pattern()
assert find_data_segment(data, "A", pattern) == '["xyz", "zyx"]'
def test_find_data_segment_with_list_of_floats_pattern_and_multiple_variables(
):
pattern = get_list_of_floats_pattern()
assert find_data_segment("A = [1]; B = [2, 3];", "B", pattern) == "[2, 3]"
def test_find_data_segment_with_list_of_floats_pattern():
pattern = get_list_of_floats_pattern()
assert find_data_segment("A = [9, 8];", "A", pattern) == "[9, 8]"
def test_find_data_segment_with_int_pattern_and_multiple_variables():
assert find_data_segment("A = 88; B = 9;", "B", get_float_pattern()) == "9"
def test_find_data_segment_with_int_pattern():
assert find_data_segment("A = 123;", "A", get_float_pattern()) == "123"