From d491ccc7bad7ce1225f9d0ca8909d43d69fdc767 Mon Sep 17 00:00:00 2001
From: Atharva Arya <aryaatharva18@moria.egr.msu.edu>
Date: Wed, 25 Dec 2024 06:17:57 -0500
Subject: [PATCH] more test files

---
 j_blue_normalized.npy                         | Bin 0 -> 1728 bytes
 j_blue_raw.npy                                | Bin 0 -> 1728 bytes
 .../montecarlo/estimators/j_blue_estimator.py |  42 ++++++++++--------
 3 files changed, 23 insertions(+), 19 deletions(-)
 create mode 100644 j_blue_normalized.npy
 create mode 100644 j_blue_raw.npy

diff --git a/j_blue_normalized.npy b/j_blue_normalized.npy
new file mode 100644
index 0000000000000000000000000000000000000000..35a082cfdbcf9a24a73059e39423c3613e576931
GIT binary patch
literal 1728
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+i=qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-2098x2AVnwwF+bcE{0Js8UmvsK(`PG{pB}z%Z55*x>-1C(P#+JFa!Xx
C0w3J~

literal 0
HcmV?d00001

diff --git a/j_blue_raw.npy b/j_blue_raw.npy
new file mode 100644
index 0000000000000000000000000000000000000000..30fdb4fbf617345f2a999e3e6422e14ba7ac40df
GIT binary patch
literal 1728
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+i=qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-2098x2AVnwwF+bcE{0Js8UmvsK(`RMS$pg2g-`3O=w{)lMWZ1=!w>)f
D<d7j<

literal 0
HcmV?d00001

diff --git a/tardis/transport/montecarlo/estimators/j_blue_estimator.py b/tardis/transport/montecarlo/estimators/j_blue_estimator.py
index 9a4e17d3425..568c4135178 100644
--- a/tardis/transport/montecarlo/estimators/j_blue_estimator.py
+++ b/tardis/transport/montecarlo/estimators/j_blue_estimator.py
@@ -19,22 +19,26 @@ def increment(self, other):
         self.j_blue_estimator += other.j_blue_estimator 
 
 if __name__ == "__main__":
-    # Create a small example with 3x3 shape
-    test_estimator = initialize_j_blue_estimator((3, 3))
-    print("Initial j_blue_estimator:")
-    print(test_estimator.j_blue_estimator)
-
-    # Create another estimator and modify it to show increment
-    other_estimator = initialize_j_blue_estimator((3, 3))
-    other_estimator.j_blue_estimator[0, 0] = 1.0
-    other_estimator.j_blue_estimator[1, 1] = 2.0
-
-    # Increment the first estimator with the second one
-    test_estimator.increment(other_estimator)
-    print("\nAfter incrementing with another estimator:")
-    print(test_estimator.j_blue_estimator)
-
-    # Save the estimator to a .npy file
-    output_file = "j_blue_estimator.npy"
-    np.save(output_file, test_estimator.j_blue_estimator)
-    print(f"\nSaved estimator to {output_file}") 
\ No newline at end of file
+    # Create example with realistic shape
+    test_estimator = initialize_j_blue_estimator((10, 20))  # typical tau_sobolev shape
+    
+    # Simulate some updates
+    packet_energy = 1.0e-10
+    packet_nu = 2.0e15
+    test_estimator.j_blue_estimator[5, 10] += packet_energy / packet_nu
+    
+    print("After simulated packet interaction:")
+    print(test_estimator.j_blue_estimator[5, 10])
+    
+    # Simulate normalization
+    time_explosion = 1.0e5
+    time_simulation = 1.0e4
+    volume = 1.0e45
+    norm_factor = 3e10 * time_explosion / (4 * np.pi * time_simulation * volume)
+    
+    normalized = test_estimator.j_blue_estimator * norm_factor
+    print("\nAfter normalization:")
+    print(normalized[5, 10])
+    
+    np.save("j_blue_raw.npy", test_estimator.j_blue_estimator)
+    np.save("j_blue_normalized.npy", normalized) 
\ No newline at end of file