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"
