File tree Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -65,7 +65,7 @@ class ExpectedResult:
65
65
def add_valid (self , list_plots : list [MockPlotInfo ]) -> None :
66
66
def create_mock_plot (info : MockPlotInfo ) -> Plot :
67
67
return Plot (
68
- info .prover .get_filename_str (),
68
+ info .prover .get_filename (),
69
69
uint8 (0 ),
70
70
bytes32 .zeros ,
71
71
None ,
@@ -77,7 +77,7 @@ def create_mock_plot(info: MockPlotInfo) -> Plot:
77
77
)
78
78
79
79
self .valid_count += len (list_plots )
80
- self .valid_delta .additions .update ({x .prover .get_filename_str (): create_mock_plot (x ) for x in list_plots })
80
+ self .valid_delta .additions .update ({x .prover .get_filename (): create_mock_plot (x ) for x in list_plots })
81
81
82
82
def remove_valid (self , list_paths : list [Path ]) -> None :
83
83
self .valid_count -= len (list_paths )
Original file line number Diff line number Diff line change 42
42
class MockDiskProver :
43
43
filename : str
44
44
45
- def get_filename (self ) -> Path :
45
+ def get_filepath (self ) -> Path :
46
46
return Path (self .filename )
47
47
48
- def get_filename_str (self ) -> str :
48
+ def get_filename (self ) -> str :
49
49
return self .filename
50
50
51
51
@@ -598,7 +598,7 @@ def assert_cache(expected: list[MockPlotInfo]) -> None:
598
598
test_cache .load ()
599
599
assert len (test_cache ) == len (expected )
600
600
for plot_info in expected :
601
- assert test_cache .get (Path (plot_info .prover .get_filename ())) is not None
601
+ assert test_cache .get (Path (plot_info .prover .get_filepath ())) is not None
602
602
603
603
# Modify two entries, with and without memo modification, they both should remain in the cache after load
604
604
modify_cache_entry (0 , 1500 , modify_memo = False )
@@ -617,7 +617,7 @@ def assert_cache(expected: list[MockPlotInfo]) -> None:
617
617
# Write the modified cache entries to the file
618
618
cache_path .write_bytes (bytes (VersionedBlob (uint16 (CURRENT_VERSION ), bytes (cache_data ))))
619
619
# And now test that plots in invalid_entries are not longer loaded
620
- assert_cache ([plot_info for plot_info in plot_infos if str (plot_info .prover .get_filename ()) not in invalid_entries ])
620
+ assert_cache ([plot_info for plot_info in plot_infos if str (plot_info .prover .get_filepath ()) not in invalid_entries ])
621
621
622
622
623
623
@pytest .mark .anyio
You can’t perform that action at this time.
0 commit comments