48
48
mock_trial_load_or_create_func ,
49
49
mock_tc_load_or_create_func ,
50
50
TEST_EXP_NAME ,
51
+ TEST_EXP_NAME_MIXED_CASE ,
51
52
TEST_RUN_NAME ,
52
53
TEST_EXP_DISPLAY_NAME ,
53
54
TEST_RUN_DISPLAY_NAME ,
@@ -779,7 +780,9 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
779
780
]
780
781
mock_tc_list .return_value = [
781
782
TrialComponentSummary (
782
- trial_component_name = Run ._generate_trial_component_name ("A" + str (i ), TEST_EXP_NAME ),
783
+ trial_component_name = Run ._generate_trial_component_name (
784
+ "A" + str (i ), TEST_EXP_NAME_MIXED_CASE
785
+ ),
783
786
trial_component_arn = "b" + str (i ),
784
787
display_name = "C" + str (i ),
785
788
source_arn = "D" + str (i ),
@@ -798,7 +801,7 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
798
801
(
799
802
_TrialComponent (
800
803
trial_component_name = Run ._generate_trial_component_name (
801
- "a" + str (i ), TEST_EXP_NAME
804
+ "a" + str (i ), TEST_EXP_NAME_MIXED_CASE
802
805
),
803
806
trial_component_arn = "b" + str (i ),
804
807
display_name = "C" + str (i ),
@@ -818,14 +821,14 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
818
821
]
819
822
820
823
run_list = list_runs (
821
- experiment_name = TEST_EXP_NAME ,
824
+ experiment_name = TEST_EXP_NAME_MIXED_CASE ,
822
825
sort_by = SortByType .CREATION_TIME ,
823
826
sort_order = SortOrderType .ASCENDING ,
824
827
sagemaker_session = sagemaker_session ,
825
828
)
826
829
827
830
mock_tc_list .assert_called_once_with (
828
- experiment_name = TEST_EXP_NAME ,
831
+ experiment_name = TEST_EXP_NAME_MIXED_CASE ,
829
832
created_before = None ,
830
833
created_after = None ,
831
834
sort_by = "CreationTime" ,
0 commit comments