def test_var_from_domain(self): d = Domain((age, gender, income), metas=(ssn, race)) self.assertEqual(d.var_from_domain(incomeA), incomeA) self.assertEqual(d.var_from_domain(incomeA, False), incomeA) with self.assertRaises(IndexError): d.var_from_domain(incomeA, True) with self.assertRaises(TypeError): d.var_from_domain(1, no_index=True) with self.assertRaises(TypeError): d.var_from_domain(-1, no_index=True)