@@ -37,55 +37,55 @@ def test_basics(self):
37
37
38
38
target_model = get_target_model ()
39
39
40
- out = editor .compare_to (model ) # Succeeds
40
+ out = editor .compare (model ) # Succeeds
41
41
self .assertEqual (out ["status" ], "success" )
42
- out = editor .compare_to (target_model ) # Fails
42
+ out = editor .compare (target_model ) # Fails
43
43
44
44
editor .add_object (
45
45
"layers/dense_3" , weights = {"0" : np .random .random ((3 , 3 ))}
46
46
)
47
- out = editor .compare_to (target_model ) # Fails
47
+ out = editor .compare (target_model ) # Fails
48
48
self .assertEqual (out ["status" ], "error" )
49
49
self .assertEqual (out ["error_count" ], 2 )
50
50
51
51
editor .rename_object ("dense_3" , "dense_4" )
52
52
editor .rename_object ("layers/dense_4" , "dense_2" )
53
53
editor .add_weights ("dense_2" , weights = {"1" : np .random .random ((3 ,))})
54
- out = editor .compare_to (target_model ) # Succeeds
54
+ out = editor .compare (target_model ) # Succeeds
55
55
self .assertEqual (out ["status" ], "success" )
56
56
57
57
editor .add_object (
58
58
"layers/dense_3" , weights = {"0" : np .random .random ((3 , 3 ))}
59
59
)
60
- out = editor .compare_to (target_model ) # Fails
60
+ out = editor .compare (target_model ) # Fails
61
61
self .assertEqual (out ["status" ], "error" )
62
62
self .assertEqual (out ["error_count" ], 1 )
63
63
64
64
editor .delete_object ("layers/dense_3" )
65
- out = editor .compare_to (target_model ) # Succeeds
65
+ out = editor .compare (target_model ) # Succeeds
66
66
self .assertEqual (out ["status" ], "success" )
67
67
editor .summary ()
68
68
69
69
temp_filepath = os .path .join (self .get_temp_dir (), "resaved.weights.h5" )
70
- editor .resave_weights (temp_filepath )
70
+ editor .save (temp_filepath )
71
71
target_model .load_weights (temp_filepath )
72
72
73
73
editor = KerasFileEditor (temp_filepath )
74
74
editor .summary ()
75
- out = editor .compare_to (target_model ) # Succeeds
75
+ out = editor .compare (target_model ) # Succeeds
76
76
self .assertEqual (out ["status" ], "success" )
77
77
78
78
editor .delete_weight ("dense_2" , "1" )
79
- out = editor .compare_to (target_model ) # Fails
79
+ out = editor .compare (target_model ) # Fails
80
80
self .assertEqual (out ["status" ], "error" )
81
81
self .assertEqual (out ["error_count" ], 1 )
82
82
83
83
editor .add_weights ("dense_2" , {"1" : np .zeros ((7 ,))})
84
- out = editor .compare_to (target_model ) # Fails
84
+ out = editor .compare (target_model ) # Fails
85
85
self .assertEqual (out ["status" ], "error" )
86
86
self .assertEqual (out ["error_count" ], 1 )
87
87
88
88
editor .delete_weight ("dense_2" , "1" )
89
89
editor .add_weights ("dense_2" , {"1" : np .zeros ((3 ,))})
90
- out = editor .compare_to (target_model ) # Succeeds
90
+ out = editor .compare (target_model ) # Succeeds
91
91
self .assertEqual (out ["status" ], "success" )
0 commit comments