import unittest
import IMP
import IMP.test
import os.path

class DirectoriesTests(IMP.test.TestCase):
    def test_data_directory(self):
        """Test key aliases"""
        k0= IMP.IntKey("0")
        k1= IMP.IntKey("1")
        ak1= IMP.IntKey.add_alias(k0, "0p")
        k2= IMP.IntKey("2")
        self.assert_(IMP.IntKey("0p") == k0)
        self.assert_(IMP.IntKey("2") == k2)
        self.assert_(k2 != ak1)

if __name__ == '__main__':
    unittest.main()