def test_descriptor_arbitrary_numpy_init(self):
"""
Test to verify creation of Descriptor object initialized
by arbitrary numpy array.
"""
descriptor_value = np.array([1, 2, 3])
descriptor = Descriptor(value=descriptor_value)
self.assertTrue(
descriptor.check_init(),
"Expected Descriptor object to be initialized",
)
self.assertEqual(
descriptor.label_,
"arbitrary",
"Expected label of descriptor initialized with "
'arbitrary vector to be "arbitrary"',
)
self.assertIsInstance(descriptor.to_numpy(), np.ndarray,
"Expected numpy.ndarray from to_numpy()")
self.assertTrue(
(descriptor.to_numpy() == descriptor_value).all(),
"Expected descriptor value to match init value",
)
with self.assertRaises(ValueError):
descriptor.to_rdkit()
def test_mordred_descriptors(self):
"""Test ability to passthrough descriptors to Mordred."""
mol_graph = MolFromSmiles(
"CC(C)C1=CC(=C(C(=C1)C(C)C)C2=CC=CC=C2P(C3CCCCC3)C4CCCCC4)C(C)C")
for desc in ["MW", "LogEE_Dt", "BalabanJ"]:
descriptor = Descriptor()
descriptor.make_fingerprint(molecule_graph=mol_graph,
fingerprint_type="mordred:" + desc)
self.assertTrue(
descriptor.check_init(),
"Expected Descriptor object to be initialized",
)
self.assertEqual(
descriptor.label_,
desc,
"Expected label of descriptor initialized with "
"{} to match the fingerprint".format(desc),
)
self.assertIsInstance(
descriptor.to_numpy(),
np.ndarray,
"Expected numpy.ndarray from to_numpy()",
)
with self.assertRaises(ValueError):
descriptor.to_rdkit()
def test_descriptor_make_fingerprint(self):
"""
Test to verify creation of Descriptor object by
creating molecular fingerprints from the molecule graph.
"""
mol_graph = MolFromSmiles("CCC")
for fprint in SUPPORTED_FPRINTS:
descriptor = Descriptor()
descriptor.make_fingerprint(molecule_graph=mol_graph,
fingerprint_type=fprint)
self.assertTrue(
descriptor.check_init(),
"Expected Descriptor object to be initialized",
)
self.assertEqual(
descriptor.label_,
fprint,
"Expected label of descriptor initialized with "
"fingerprint to match the fingerprint",
)
self.assertIsInstance(
descriptor.to_numpy(),
np.ndarray,
"Expected numpy.ndarray from to_numpy()",
)
self.assertIsInstance(
descriptor.to_rdkit(),
ExplicitBitVect,
"Expected to_rdkit() to return "
"ExplicitBitVect representation "
f"of {fprint} fingerprint",
)
def test_padelpy_descriptors(self):
"""Test ability to passthrough descriptors to PadelPy."""
mol_graph = MolFromSmiles("CCOCC")
for desc in ["MATS7e", "Ti", "ATSC6p"]:
descriptor = Descriptor()
descriptor.make_fingerprint(molecule_graph=mol_graph,
fingerprint_type="padelpy:" + desc)
self.assertTrue(
descriptor.check_init(),
"Expected Descriptor object to be initialized",
)
self.assertEqual(
descriptor.label_,
desc,
"Expected label of descriptor initialized with "
"{} to match the fingerprint".format(desc),
)
self.assertIsInstance(
descriptor.to_numpy(),
np.ndarray,
"Expected numpy.ndarray from to_numpy()",
)
with self.assertRaises(ValueError):
descriptor.to_rdkit()
