import IMP
import IMP.test

class RandomTests(IMP.test.TestCase):
    def test_random_number_generator(self):
        """Test the random number generator"""
        IMP.random_number_generator.seed(24)
        a1 = IMP.random_number_generator()
        a2 = IMP.random_number_generator()
        # In principle these values could be the same, I suppose, but it is
        # rather unlikely!
        self.assertNotEqual(a1, a2)
        # Reseeding should give a predictable sequence
        IMP.random_number_generator.seed(24)
        a3 = IMP.random_number_generator()
        a4 = IMP.random_number_generator()
        self.assertEqual(a1, a3)
        self.assertEqual(a2, a4)

if __name__ == '__main__':
    IMP.test.main()