def test_get_reference_type(self): ds = DataStore('testing/data/reference_types') mt1 = ds.get_type('Type1') self.assertEqual('testing.reference_types.Type1', mt1.name) self.assertEqual('Type1', mt1.ref_type) mt2 = ds.get_type('Type2') self.assertEqual('testing.reference_types.Type2', mt2.name) self.assertEqual('Type2', mt2.ref_type) with self.assertRaises(NotAManagedType): ds.get_type('XXX')