mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-05 22:04:27 +08:00
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
from ML.FeatureSelect import CMIM
|
|
import DataStructs as DS
|
|
import RDConfig
|
|
import unittest
|
|
|
|
class TestCase(unittest.TestCase):
|
|
def setUp(self) :
|
|
pass
|
|
|
|
def test0FromList(self) :
|
|
examples = []
|
|
|
|
bv = DS.ExplicitBitVect(5)
|
|
bv.SetBitsFromList([0,2,4])
|
|
examples.append([0,bv,0])
|
|
|
|
bv = DS.ExplicitBitVect(5)
|
|
bv.SetBitsFromList([0,2,4])
|
|
examples.append([0,bv,0])
|
|
|
|
bv = DS.ExplicitBitVect(5)
|
|
bv.SetBitsFromList([0,3,4])
|
|
examples.append([0,bv,1])
|
|
|
|
bv = DS.ExplicitBitVect(5)
|
|
bv.SetBitsFromList([0,2,4])
|
|
examples.append([0,bv,0])
|
|
|
|
bv = DS.ExplicitBitVect(5)
|
|
bv.SetBitsFromList([0,2])
|
|
examples.append([0,bv,1])
|
|
|
|
r = CMIM.SelectFeatures(examples,2)
|
|
self.failUnless(r==(2,4))
|
|
|
|
# here we ask for three features, but there are only two
|
|
# that are non-redundant:
|
|
r = CMIM.SelectFeatures(examples,3)
|
|
self.failUnless(r==(2,4))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|