Ejemplo n.º 1
0
for t1, t2 in zip(sp1.tallies, sp2.tallies):
    # Compare size of tallies
    assert t1.total_score_bins == t2.total_score_bins
    assert t1.total_filter_bins == t2.total_filter_bins

    # Compare filters
    assert len(t1.filters) == len(t2.filters)
    for f1, f2 in zip(t1.filters.values(), t2.filters.values()):
        assert f1.type == f2.type
        assert f1.length == f2.length
        assert_equal(f1.bins, f2.bins)

    # Compare nuclide and score bins
    assert_equal(t1.nuclides, t2.nuclides)
    assert_equal(t1.scores, t2.scores)

    # Compare tally results
    assert_allclose(t1.results, t2.results)

# If criticality, compare source sites
if sp1.run_mode == 2:
    sp1.read_source()
    sp2.read_source()

    assert len(sp1.source) == len(sp2.source)
    for s1, s2 in zip(sp1.source, sp2.source):
        assert s1.weight == s2.weight
        assert_allclose(s1.xyz, s2.xyz)
        assert_allclose(s1.uvw, s2.uvw)
        assert s1.E == s2.E