def test_cadis_multiple_e(): """Test multiple energy group CADIS case""" adj_flux_tag = "adj_flux" q_tag = "q" ww_tag = "ww" q_bias_tag= "q_bias" #create meshes coords = [[0, 1, 2], [-1, 3, 4], [10, 12]] adj_flux_mesh = Mesh(structured=True, structured_coords=coords) q_mesh = Mesh(structured=True, structured_coords=coords) ww_mesh = Mesh(structured=True, structured_coords=coords) q_bias_mesh = Mesh(structured=True, structured_coords=coords) #add tags to input meshes tag_adj_flux = adj_flux_mesh.mesh.createTag(adj_flux_tag, 2, float) tag_q = q_mesh.mesh.createTag(q_tag, 2, float) #create data for input meshes adj_flux_data = [[1.1, 1.2], [1.3, 1.4], [1.5, 1.6], [1.7, 1.8]] q_data = [[2.9, 2.8], [2.6, 2.5], [2.4, 2.2], [2.2, 2.0]] #data data to mesh adj_flux_ves = list(adj_flux_mesh.mesh.iterate(iBase.Type.region, iMesh.Topology.all)) for i, adj_flux_ve in enumerate(adj_flux_ves): tag_adj_flux[adj_flux_ve] = adj_flux_data[i] q_ves = list(q_mesh.mesh.iterate(iBase.Type.region, iMesh.Topology.all)) for i, q_ve in enumerate(q_ves): tag_q[q_ve] = q_data[i] #run cadis cadis(adj_flux_mesh, adj_flux_tag, q_mesh, q_tag, ww_mesh, ww_tag, q_bias_mesh, q_bias_tag, beta=5) #expected results expected_ww = [[1.4535005225, 1.3869047619], [1.3717948718, 1.3314285714], [1.287962963, 1.3238636364], [1.2397504456, 1.2944444444]] expected_q_bias = [[0.2293314162, 0.2403433476], [0.2429906542, 0.2503576538], [0.2588066139, 0.251788269], [0.2688713156, 0.2575107296]] ww_ves = list(ww_mesh.mesh.iterate(iBase.Type.region, iMesh.Topology.all)) for i, ww_ve in enumerate(ww_ves): assert_array_almost_equal(ww_mesh.mesh.getTagHandle(ww_tag)[ww_ve], expected_ww[i]) q_bias_ves = list(q_bias_mesh.mesh.iterate(iBase.Type.region, iMesh.Topology.all)) for i, q_bias_ve in enumerate(q_bias_ves): assert_array_almost_equal( q_bias_mesh.mesh.getTagHandle(q_bias_tag)[q_bias_ve], expected_q_bias[i])
def test_cadis_multiple_e(): """Test multiple energy group CADIS case""" adj_flux_tag = "adj_flux" q_tag = "q" ww_tag = "ww" q_bias_tag = "q_bias" # create meshes coords = [[0, 1, 2], [-1, 3, 4], [10, 12]] adj_flux_mesh = Mesh(structured=True, structured_coords=coords) q_mesh = Mesh(structured=True, structured_coords=coords) ww_mesh = Mesh(structured=True, structured_coords=coords) q_bias_mesh = Mesh(structured=True, structured_coords=coords) # add tags to input meshes adj_flux_mesh.adj_flux = NativeMeshTag(2, float) q_mesh.q = NativeMeshTag(2, float) # create data for input meshes adj_flux_data = [[1.1, 1.2], [1.3, 1.4], [0.0, 1.6], [1.7, 1.9]] q_data = [[2.9, 2.8], [2.6, 2.5], [2.4, 2.2], [2.9, 0.0]] # data data to mesh adj_flux_mesh.adj_flux[:] = adj_flux_data q_mesh.q[:] = q_data # run cadis cadis( adj_flux_mesh, adj_flux_tag, q_mesh, q_tag, ww_mesh, ww_tag, q_bias_mesh, q_bias_tag, beta=5, ) # expected results expected_q_bias = [ [0.0306200806, 0.0322518718], [0.0324438472, 0.0335956998], [0.0, 0.0337876752], [0.0473219428, 0.0], ] expected_ww = [ [0.3208302538, 0.2940943993], [0.2714717532, 0.2520809137], [0.0, 0.2205707995], [0.2075960465, 0.1857438311], ] ww_mesh.ww = NativeMeshTag(2, float) q_bias_mesh.q_bias = NativeMeshTag(2, float) assert_array_almost_equal(ww_mesh.ww[:], expected_ww[:]) assert_array_almost_equal(q_bias_mesh.q_bias[:], expected_q_bias[:])
def test_cadis_single_e(): """Test single energy group cadis case""" adj_flux_tag = "adj_flux" q_tag = "q" ww_tag = "ww" q_bias_tag= "q_bias" #create meshes coords = [[0, 1, 2], [-1, 3, 4], [10, 12]] adj_flux_mesh = Mesh(structured=True, structured_coords=coords) q_mesh = Mesh(structured=True, structured_coords=coords) ww_mesh = Mesh(structured=True, structured_coords=coords) q_bias_mesh = Mesh(structured=True, structured_coords=coords) #add tags to input meshes tag_adj_flux = adj_flux_mesh.mesh.createTag(adj_flux_tag, 1, float) tag_q = q_mesh.mesh.createTag(q_tag, 1, float) #create data for input meshes adj_flux_data = [1.1, 1.3, 1.5, 1.7] q_data = [2.9, 2.6, 2.4, 2.2] #data data to mesh adj_flux_ves = list(adj_flux_mesh.mesh.iterate(iBase.Type.region, iMesh.Topology.all)) for i, adj_flux_ve in enumerate(adj_flux_ves): tag_adj_flux[adj_flux_ve] = adj_flux_data[i] q_ves = list(q_mesh.mesh.iterate(iBase.Type.region, iMesh.Topology.all)) for i, q_ve in enumerate(q_ves): tag_q[q_ve] = q_data[i] #run CADIS cadis(adj_flux_mesh, adj_flux_tag, q_mesh, q_tag, ww_mesh, ww_tag, q_bias_mesh, q_bias_tag, beta=5) #checkout output meshes expected_ww = [1.4535005225, 1.3717948718, 1.287962963, 1.2397504456] expected_q_bias = [0.2293314162, 0.2429906542, 0.2588066139, 0.2688713156] ww_ves = list(ww_mesh.mesh.iterate(iBase.Type.region, iMesh.Topology.all)) for i, ww_ve in enumerate(ww_ves): assert_almost_equal(ww_mesh.mesh.getTagHandle(ww_tag)[ww_ve], expected_ww[i]) q_bias_ves = list(q_bias_mesh.mesh.iterate(iBase.Type.region, iMesh.Topology.all)) for i, q_bias_ve in enumerate(q_bias_ves): assert_almost_equal( q_bias_mesh.mesh.getTagHandle(q_bias_tag)[q_bias_ve], expected_q_bias[i])
def test_cadis_single_e(): """Test single energy group cadis case""" adj_flux_tag = "adj_flux" q_tag = "q" ww_tag = "ww" q_bias_tag = "q_bias" # create meshes coords = [[0, 1, 2], [-1, 3, 4], [10, 12]] adj_flux_mesh = Mesh(structured=True, structured_coords=coords) q_mesh = Mesh(structured=True, structured_coords=coords) ww_mesh = Mesh(structured=True, structured_coords=coords) q_bias_mesh = Mesh(structured=True, structured_coords=coords) # add tags to meshes adj_flux_mesh.adj_flux = NativeMeshTag(1, float) q_mesh.q = NativeMeshTag(1, float) # create data for input meshes adj_flux_data = [1.1, 1.3, 1.5, 1.7] q_data = [2.9, 2.6, 2.4, 2.2] # data data to mesh adj_flux_mesh.adj_flux[:] = adj_flux_data q_mesh.q[:] = q_data # run CADIS cadis( adj_flux_mesh, adj_flux_tag, q_mesh, q_tag, ww_mesh, ww_tag, q_bias_mesh, q_bias_tag, beta=5, ) # checkout output meshes expected_ww = [0.3995338, 0.33806706, 0.29299145, 0.258521908] expected_q_bias = [0.04652859, 0.04929988, 0.052508751, 0.0545507858] ww_mesh.ww = NativeMeshTag(1, float) q_bias_mesh.q_bias = NativeMeshTag(1, float) assert_array_almost_equal(ww_mesh.ww[:], expected_ww[:]) assert_array_almost_equal(q_bias_mesh.q_bias[:], expected_q_bias[:])
def test_cadis_multiple_e(): """Test multiple energy group CADIS case""" adj_flux_tag = "adj_flux" q_tag = "q" ww_tag = "ww" q_bias_tag= "q_bias" #create meshes coords = [[0, 1, 2], [-1, 3, 4], [10, 12]] adj_flux_mesh = Mesh(structured=True, structured_coords=coords) q_mesh = Mesh(structured=True, structured_coords=coords) ww_mesh = Mesh(structured=True, structured_coords=coords) q_bias_mesh = Mesh(structured=True, structured_coords=coords) #add tags to input meshes adj_flux_mesh.adj_flux = IMeshTag(2, float) q_mesh.q = IMeshTag(2, float) #create data for input meshes adj_flux_data = [[1.1, 1.2], [1.3, 1.4], [0.0, 1.6], [1.7, 1.9]] q_data = [[2.9, 2.8], [2.6, 2.5], [2.4, 2.2], [2.9, 0.0]] #data data to mesh adj_flux_mesh.adj_flux[:] = adj_flux_data q_mesh.q[:] = q_data #run cadis cadis(adj_flux_mesh, adj_flux_tag, q_mesh, q_tag, ww_mesh, ww_tag, q_bias_mesh, q_bias_tag, beta=5) #expected results expected_q_bias = [[0.0306200806, 0.0322518718], [0.0324438472, 0.0335956998], [0.0, 0.0337876752], [0.0473219428, 0.0]] expected_ww = [[0.3208302538, 0.2940943993], [0.2714717532, 0.2520809137], [0.0, 0.2205707995], [0.2075960465, 0.1857438311]] ww_mesh.ww = IMeshTag(2, float) q_bias_mesh.q_bias = IMeshTag(2, float) assert_array_almost_equal(ww_mesh.ww[:], expected_ww[:]) assert_array_almost_equal(q_bias_mesh.q_bias[:], expected_q_bias[:])
def test_cadis_single_e(): """Test single energy group cadis case""" adj_flux_tag = "adj_flux" q_tag = "q" ww_tag = "ww" q_bias_tag= "q_bias" #create meshes coords = [[0, 1, 2], [-1, 3, 4], [10, 12]] adj_flux_mesh = Mesh(structured=True, structured_coords=coords) q_mesh = Mesh(structured=True, structured_coords=coords) ww_mesh = Mesh(structured=True, structured_coords=coords) q_bias_mesh = Mesh(structured=True, structured_coords=coords) #add tags to meshes adj_flux_mesh.adj_flux = IMeshTag(1, float) q_mesh.q = IMeshTag(1, float) #create data for input meshes adj_flux_data = [1.1, 1.3, 1.5, 1.7] q_data = [2.9, 2.6, 2.4, 2.2] #data data to mesh adj_flux_mesh.adj_flux[:] = adj_flux_data q_mesh.q[:] = q_data #run CADIS cadis(adj_flux_mesh, adj_flux_tag, q_mesh, q_tag, ww_mesh, ww_tag, q_bias_mesh, q_bias_tag, beta=5) #checkout output meshes expected_ww = [0.3995338, 0.33806706, 0.29299145, 0.258521908] expected_q_bias = [0.04652859, 0.04929988, 0.052508751, 0.0545507858] ww_mesh.ww = IMeshTag(1, float) q_bias_mesh.q_bias = IMeshTag(1, float) assert_array_almost_equal(ww_mesh.ww[:], expected_ww[:]) assert_array_almost_equal(q_bias_mesh.q_bias[:], expected_q_bias[:])