diff --git a/generate_dummy_data.py b/generate_dummy_data.py index a24bbc8..e764ed8 100644 --- a/generate_dummy_data.py +++ b/generate_dummy_data.py @@ -13,14 +13,20 @@ processed_path = os.path.join(base_path, "processed") os.makedirs(raw_path, exist_ok=True) os.makedirs(processed_path, exist_ok=True) -for damage in range(1, 6): # 5 Damage levels - damage_folder = f"DAMAGE_{damage}" - damage_path = os.path.join(processed_path, damage_folder) +# Define the number of zeros to pad +num_damages = 5 +num_tests = 10 +damage_pad = len(str(num_damages)) +test_pad = len(str(num_tests)) + +for damage in range(1, num_damages + 1): # 5 Damage levels starts from 1 + damage_folder = f"DAMAGE_{damage:0{damage_pad}}" + damage_path = os.path.join(raw_path, damage_folder) os.makedirs(damage_path, exist_ok=True) for test in range(1, 11): # 10 Tests per damage level # Filename for the CSV - csv_filename = f"D{damage}_TEST{test}.csv" + csv_filename = f"D{damage:0{damage_pad}}_TEST{test:0{test_pad}}.csv" csv_path = os.path.join(damage_path, csv_filename) # Generate dummy data