def test_mixture_prob(): obs = '1061313128_99bl_1pol_half_time' testfile = os.path.join(DATA_PATH, '%s.uvfits' % obs) file_type = 'uvfits' ss = SS() ss.read(testfile, diff=True) ss.apply_flags('original') # Generate the mixture probabilities mixture_prob = ss.mixture_prob(bins='auto') # Check that they sum to close to 1 assert np.isclose(np.sum(mixture_prob), 1), "Probabilities did not add up to close to 1" # Do a new read, but don't diff. Run and check mask. ss = SS() ss.read(testfile, diff=False) mixture_prob = ss.mixture_prob(bins='auto') assert ss.flag_choice is None
def test_mixture_prob(): obs = '1061313128_99bl_1pol_half_time' testfile = os.path.join(DATA_PATH, '%s.uvfits' % obs) file_type = 'uvfits' ss = SS() ss.read(testfile) ss.apply_flags('original') # Generate the mixture probabilities mixture_prob = ss.mixture_prob(bins='auto') # Check that they sum to close to 1 assert np.isclose(np.sum(mixture_prob), 1), "Probabilities did not add up to close to 1"