from ... import lamda

def test_query():
    Q = lamda.core.LAMDAQuery()
    Q.lamda_query(mol='co', query_type='erg_levels')
    Q.lamda_query(mol='co', query_type='rad_trans')
    Q.lamda_query(mol='co', query_type='coll_rates')


from ... import lamda

def test_query():
    lamda.print_mols()
    lamda.query(mol='co', query_type='erg_levels')
    lamda.query(mol='co', query_type='rad_trans')
    lamda.query(mol='co', query_type='coll_rates', coll_partner_index=1)


Update tests for new style
Update tests for new style

Also added test for printing molecule list and made the collisional rate
test more complicated.

  from ... import lamda
  
  def test_query():
-     Q = lamda.core.LAMDAQuery()
+     lamda.print_mols()
-     Q.lamda_query(mol='co', query_type='erg_levels')
+     lamda.query(mol='co', query_type='erg_levels')
-     Q.lamda_query(mol='co', query_type='rad_trans')
+     lamda.query(mol='co', query_type='rad_trans')
-     Q.lamda_query(mol='co', query_type='coll_rates')
+     lamda.query(mol='co', query_type='coll_rates', coll_partner_index=1)
  
  
Update tests for new style
## Code Before:
from ... import lamda

def test_query():
    Q = lamda.core.LAMDAQuery()
    Q.lamda_query(mol='co', query_type='erg_levels')
    Q.lamda_query(mol='co', query_type='rad_trans')
    Q.lamda_query(mol='co', query_type='coll_rates')


## Instruction:
Update tests for new style
## Code After:
from ... import lamda

def test_query():
    lamda.print_mols()
    lamda.query(mol='co', query_type='erg_levels')
    lamda.query(mol='co', query_type='rad_trans')
    lamda.query(mol='co', query_type='coll_rates', coll_partner_index=1)

