from __future__ import division from iotbx.pdb.multimer_reconstruction import multimer from iotbx.ncs.ncs_preprocess import format_num_as_str from iotbx.ncs.ncs_preprocess import ncs_group_object from mmtbx.ncs.ncs_utils import apply_transforms from libtbx.test_utils import approx_equal from scitbx.array_family import flex from scitbx import matrix from iotbx import pdb import iotbx.ncs import unittest import tempfile import shutil import os import sys class TestMultimerReconstruction(unittest.TestCase): '''Test multimer reconstruction from BIOMT and MTRIX records''' def setUp(self): ''' Create temporary file: multimer_test_data.pdb ''' self.currnet_dir = os.getcwd() self.tempdir = tempfile.mkdtemp('tempdir') os.chdir(self.tempdir) # Write the test data into a file open('multimer_test_data.pdb', 'w').write(pdb_test_data) open('multimer_test_data2.pdb', 'w').write(pdb_test_data2) # @unittest.SkipTest def test_MTRIX(self): '''Test MTRIX record processing''' print sys._getframe().f_code.co_name cau_expected_results = [ [1.0, 1.0, 1.0], [1.0, -1.0, 1.0], [-0.366025, 1.366025, 1.0], [-1.366025, 0.366025, 1.0], [1.0, 1.5, 1.0], [94.618, -5.253, 91.582], [94.618, -91.582, -5.253], [51.858229, 79.315053, 91.582], [-42.759771, 84.568053, 91.582], [94.618, -4.753, 91.582], [62.395, 51.344, 80.786], [62.395, -80.786, 51.344], [-13.267688, 79.70763, 80.786], [-75.662688, 28.36363, 80.786], [62.395, 51.844, 80.786], [39.954, 51.526, 72.372], [39.954, -72.372, 51.526], [-24.645804, 60.364163, 72.372], [-64.599804, 8.838163, 72.372], [39.954, 52.026, 72.372]] # use MTRIX data multimer_data = multimer( file_name='multimer_test_data.pdb', reconstruction_type='cau') cau_multimer_xyz = list(multimer_data.sites_cart()) cau_multimer_xyz.sort() cau_expected_results.sort() assert approx_equal(cau_expected_results,cau_multimer_xyz,eps=0.001) # Test that the number of MTRIX record to process is correct self.assertEqual(multimer_data.number_of_transforms,4) # Test getting non-rounded ASU source_xyz = multimer_data.get_ncs_hierarchy().atoms().extract_xyz() xyz = apply_transforms( ncs_coordinates = source_xyz, ncs_restraints_group_list = multimer_data.get_ncs_restraints_group_list(), total_asu_length = multimer_data.total_asu_length(), extended_ncs_selection = flex.size_t_range(source_xyz.size()), round_coordinates=False) cau_multimer_xyz = list(xyz) cau_multimer_xyz.sort() assert approx_equal(cau_expected_results,cau_multimer_xyz,eps=0.00001) # Test multimer without rounding multimer_data = multimer( file_name='multimer_test_data.pdb', round_coordinates=False, reconstruction_type='cau') cau_multimer_xyz = list(multimer_data.sites_cart()) cau_multimer_xyz.sort() cau_expected_results.sort() assert approx_equal(cau_expected_results,cau_multimer_xyz,eps=0.00001) # @unittest.SkipTest def test_BIOMT(self): '''Test MTRIX record processing''' print sys._getframe().f_code.co_name ba_expected_results = [ [1.0, 1.0, 1.0], [1.0, -1.0, 1.0], [1.0, 1.0, -1.0], [-1.0, 1.0, 1.0], [1.0, 1.0, -1.0], [-1.0, 1.0, -1.0], [-0.366025, 1.366025, 1.0], [-1.366025, 0.366025, 1.0], [1.0, 1.5, 1.0], [-1.366025, 0.366025, 0.0], [94.618, -5.253, 91.582], [94.618, -91.582, -5.253], [91.582, -5.253, -94.618], [5.253, 94.618, 91.582], [91.582, -5.253, -94.618], [5.253, 91.582, -94.618], [51.858229, 79.315053, 91.582], [-42.759771, 84.568053, 91.582], [94.618, -4.753, 91.582], [-42.759771, 84.568053, 90.582], [62.395, 51.344, 80.786], [62.395, -80.786, 51.344], [80.786, 51.344, -62.395], [-51.344, 62.395, 80.786], [80.786, 51.344, -62.395], [-51.344, 80.786, -62.395], [-13.267688, 79.70763, 80.786], [-75.662688, 28.36363, 80.786], [62.395, 51.844, 80.786], [-75.662688, 28.36363, 79.786], [39.954, 51.526, 72.372], [39.954, -72.372, 51.526], [72.372, 51.526, -39.954], [-51.526, 39.954, 72.372], [72.372, 51.526, -39.954], [-51.526, 72.372, -39.954], [-24.645804, 60.364163, 72.372], [-64.599804, 8.838163, 72.372], [39.954, 52.026, 72.372], [-64.599804, 8.838163, 71.372]] # use BIOMT data ba_multimer_data = multimer( file_name='multimer_test_data.pdb', reconstruction_type='ba') ba_multimer_xyz = list(ba_multimer_data.sites_cart()) # The multimer processing is my chain, so we need to sort lists before we compare them ba_multimer_xyz.sort() ba_expected_results.sort() assert approx_equal(ba_expected_results,ba_multimer_xyz,eps=0.001) self.assertEqual(ba_multimer_data.number_of_transforms,9) # @unittest.SkipTest def test_ncs_copies_naming(self): print sys._getframe().f_code.co_name transforms_obj = ncs_group_object() result = transforms_obj.make_chains_names( ['chain A_001','chain B_001','chain A_002','chain B_002'],('A','B')) expected = {'chain A_001': 'C', 'chain B_001': 'D', 'chain A_002': 'E', 'chain B_002': 'F'} self.assertEqual(result,expected) def test_identity_tranform_insertion(self): """ Verify that insertion and reordering of the identity transform is done properly """ print sys._getframe().f_code.co_name for pdb_str in [pdb_test_data5,pdb_test_data6]: ncs_inp = iotbx.ncs.input(pdb_string=pdb_test_data5) transform_info = ncs_inp.build_MTRIX_object() self.assertEqual(len(transform_info.r),3) self.assertEqual(len(transform_info.t),3) self.assertEqual(transform_info.r[0].is_r3_identity_matrix(),True) self.assertEqual(transform_info.t[0].is_col_zero(),True) sn = [int(x) for x in transform_info.serial_number] self.assertEqual(sn,[1,2,3]) # @unittest.SkipTest def test_adding_transforms_directly(self): """ Verify that processing of transforms provided manually is done properly """ print sys._getframe().f_code.co_name pdb_obj = pdb.hierarchy.input(pdb_string=pdb_test_data4) r = [matrix.sqr([0.1,1.0,1.0,0.2,0.5,0.6,0.7,0.8,0.9])] r.append(matrix.sqr([1.0,0.2,1.0,0.2,0.5,0.6,0.7,0.8,0.4])) t = [matrix.col([0,2,1])] t.append(matrix.col([-1,3,-2])) transforms_obj = iotbx.ncs.input( pdb_hierarchy_inp = pdb_obj, rotations=r, translations=t) result = transforms_obj.transform_to_ncs expected = {'002': ['chain A_002', 'chain B_002'], '003': ['chain A_003', 'chain B_003']} self.assertEqual(result,expected) result = transforms_obj.ncs_selection_str expected = 'chain A or chain B' self.assertEqual(result,expected) # check that if transforms are provided MTRIX record ignored pdb_obj = pdb.hierarchy.input(pdb_string=pdb_test_data2) transforms_obj = iotbx.ncs.input( pdb_hierarchy_inp = pdb_obj, rotations=r, translations=t) result = transforms_obj.transform_to_ncs expected = {'002': ['chain A_002', 'chain B_002'], '003': ['chain A_003', 'chain B_003']} self.assertEqual(result,expected) result = transforms_obj.ncs_selection_str expected = 'chain A or chain B' self.assertEqual(result,expected) # transforms that are not present result = transforms_obj.transform_to_ncs.keys() expected = ['003', '002'] self.assertEqual(result,expected) # all transforms result = transforms_obj.ncs_transform.keys() expected = ['001', '002', '003'] result.sort() self.assertEqual(result,expected) # @unittest.SkipTest def test_transform_application_order(self): """ Verify that transform order is kept even when chain selection is complex """ print sys._getframe().f_code.co_name pdb_inp = pdb.input(source_info=None, lines=pdb_test_data2) pdb_obj = pdb.hierarchy.input(pdb_string=pdb_test_data2) transform_info = pdb_inp.process_mtrix_records() transforms_obj = iotbx.ncs.input( transform_info=transform_info, pdb_hierarchy_inp=pdb_obj) expected = ['chain A_002', 'chain B_002', 'chain A_003', 'chain B_003'] self.assertEqual(transforms_obj.transform_chain_assignment,expected) expected = { 'chain A_002': 'C','chain A_003': 'E','chain A_001': 'A', 'chain B_002': 'D','chain B_003': 'F','chain B_001': 'B'} self.assertEqual(transforms_obj.ncs_copies_chains_names,expected) expected = [0, 1, 2, 5, 6] results = list(transforms_obj.asu_to_ncs_map['chain A']) self.assertEqual(results,expected) expected = [3, 4] results = list(transforms_obj.asu_to_ncs_map['chain B']) self.assertEqual(results,expected) expected = [7, 8, 9, 12, 13] results = list(transforms_obj.ncs_to_asu_map['chain A_002']) self.assertEqual(results,expected) expected = [17, 18] results = list(transforms_obj.ncs_to_asu_map['chain B_003']) self.assertEqual(results,expected) self.assertEqual(len(transforms_obj.ncs_atom_selection),21) self.assertEqual(transforms_obj.ncs_atom_selection.count(True),7) # @unittest.SkipTest def test_num_to_str(self): print sys._getframe().f_code.co_name self.assertEqual(format_num_as_str(946),'946') self.assertEqual(format_num_as_str(46),'046') self.assertEqual(format_num_as_str(6),'006') self.assertEqual(format_num_as_str(0),'000') def test_pdb_writing(self): print sys._getframe().f_code.co_name """ Verify that there are no errors processing the write command No inception of the output is done. To view the output change the write=False to ,write=True """ transforms_obj = ncs_group_object() pdb_inp = pdb.input(source_info=None, lines=pdb_test_data2) pdb_obj = pdb.hierarchy.input(pdb_string=pdb_test_data2) transform_info = pdb_inp.process_mtrix_records() transforms_obj.preprocess_ncs_obj( transform_info=transform_info, pdb_hierarchy_inp=pdb_obj) multimer_data = multimer( pdb_str=pdb_test_data2, reconstruction_type='cau') pdb_hierarchy_asu = multimer_data.assembled_multimer # print '--- using ASU hierarchy ---' pdbstr = transforms_obj.get_transform_records( ncs_only=True, pdb_hierarchy=pdb_hierarchy_asu, write=False) # print pdbstr # print '='*50 pdbstr = transforms_obj.get_transform_records( ncs_only=False, pdb_hierarchy=pdb_hierarchy_asu, write=False) # print pdbstr # print '--- using the hierarchy of only the master NCS ---' pdbstr = transforms_obj.get_transform_records( pdb_hierarchy=pdb_obj.hierarchy, biomt=True, write=False) # print pdbstr # print '--- from xray structure ---' xrs = pdb_hierarchy_asu.extract_xray_structure() pdbstr = transforms_obj.get_transform_records( # xrs=pdb_obj.xray_structure_simple(), xrs=xrs, biomt=True, write=False) # print pdbstr # print '='*50 def test_spec_file_format(self): """ Verify that spec object are produced properly """ print sys._getframe().f_code.co_name multimer_data = multimer( pdb_str=pdb_test_data2, reconstruction_type='cau') trans_obj = iotbx.ncs.input(pdb_string=pdb_test_data2) pdb_hierarchy_asu = multimer_data.assembled_multimer spec_output = trans_obj.get_ncs_info_as_spec( pdb_hierarchy_asu=pdb_hierarchy_asu,write=False) trans_obj2 = iotbx.ncs.input(spec_ncs_groups=spec_output) t1 = trans_obj.ncs_transform['002'].r t2 = trans_obj2.ncs_transform['002'].r self.assertEqual(t1,t2) self.assertEqual(len(trans_obj.ncs_transform),len(trans_obj2.ncs_transform)) t1 = trans_obj.ncs_to_asu_selection t1_expected = {'chain A': ['chain C', 'chain E'], 'chain B': ['chain D', 'chain F']} self.assertEqual(t1,t1_expected) t2 = trans_obj2.ncs_to_asu_selection t2_expected = { 'chain A and (resseq 1:3 or resseq 6:7)': ['chain C and (resseq 1:3 or resseq 6:7)', 'chain E and (resseq 1:3 or resseq 6:7)'], 'chain B and (resseq 4:5)': ['chain D and (resseq 4:5)', 'chain F and (resseq 4:5)']} self.assertEqual(t2,t2_expected) t1 = trans_obj.tr_id_to_selection['chain A_003'] t1_expected = ('chain A', 'chain E') self.assertEqual(t1,t1_expected) t2 = trans_obj2.tr_id_to_selection['chain A_003'] t2_expected = ('chain A and (resseq 1:3 or resseq 6:7)', 'chain E and (resseq 1:3 or resseq 6:7)') self.assertEqual(t2,t2_expected) # @unittest.SkipTest def test_writing_spec_file(self): print sys._getframe().f_code.co_name """ Verify that there are no errors processing the write command No inception of the output is done. Just making sure it does not break To view the output change the write=False to ,write=True """ transforms_obj = ncs_group_object() pdb_inp = pdb.input(source_info=None, lines=pdb_test_data2) pdb_obj = pdb.hierarchy.input(pdb_string=pdb_test_data2) transform_info = pdb_inp.process_mtrix_records() transforms_obj.preprocess_ncs_obj( # transform_info=transform_info, # pdb_hierarchy_inp=pdb_obj, pdb_inp=pdb_inp) asu = multimer( pdb_str=pdb_test_data2, reconstruction_type='cau') transforms_obj.get_ncs_info_as_spec( pdb_hierarchy_asu=asu.assembled_multimer,write=False) def tearDown(self): '''remove temp files and folder''' os.chdir(self.currnet_dir) shutil.rmtree(self.tempdir) pdb_test_data = '''\ REMARK 0 Test molecule with BIOMOLECULE: 1 REMARK 0 REMARK 0 The test will generate the biomolecule (the multimer assembly) REMARK 0 from the transformation matrices writen below REMARK 0 and then compare the results to the calculated expected one REMARK 350 CRYSTALLOGRAPHIC OPERATIONS ARE GIVEN. REMARK 350 BIOMT1 1 1.000000 0.000000 0.000000 0.000000 REMARK 350 BIOMT2 1 0.000000 1.000000 0.000000 0.000000 REMARK 350 BIOMT3 1 0.000000 0.000000 1.000000 0.000000 REMARK 350 BIOMT1 2 1.000000 0.000000 0.000000 0.000000 REMARK 350 BIOMT2 2 0.000000 0.000000 -1.000000 0.000000 REMARK 350 BIOMT3 2 0.000000 1.000000 0.000000 0.000000 REMARK 350 BIOMT1 3 0.000000 0.000000 1.000000 0.000000 REMARK 350 BIOMT2 3 0.000000 1.000000 0.000000 0.000000 REMARK 350 BIOMT3 3 -1.000000 0.000000 0.000000 0.000000 REMARK 350 BIOMT1 4 0.000000 -1.000000 0.000000 0.000000 REMARK 350 BIOMT2 4 1.000000 0.000000 0.000000 0.000000 REMARK 350 BIOMT3 4 0.000000 0.000000 1.000000 0.000000 REMARK 350 BIOMT1 5 0.000000 0.000000 1.000000 0.000000 REMARK 350 BIOMT2 5 0.000000 1.000000 0.000000 0.000000 REMARK 350 BIOMT3 5 -1.000000 0.000000 0.000000 0.000000 REMARK 350 BIOMT1 6 0.000000 -1.000000 0.000000 0.000000 REMARK 350 BIOMT2 6 0.000000 0.000000 1.000000 0.000000 REMARK 350 BIOMT3 6 -1.000000 0.000000 0.000000 0.000000 REMARK 350 BIOMT1 7 0.500000 -0.866025 0.000000 0.000000 REMARK 350 BIOMT2 7 0.866025 0.500000 0.000000 0.000000 REMARK 350 BIOMT3 7 0.000000 0.000000 1.000000 0.000000 REMARK 350 BIOMT1 8 -0.500000 -0.866025 0.000000 0.000000 REMARK 350 BIOMT2 8 0.866025 -0.500000 0.000000 0.000000 REMARK 350 BIOMT3 8 0.000000 0.000000 1.000000 0.000000 REMARK 350 BIOMT1 9 1.000000 0.000000 0.000000 0.000000 REMARK 350 BIOMT2 9 0.000000 1.000000 0.000000 0.500000 REMARK 350 BIOMT3 9 0.000000 0.000000 1.000000 0.000000 REMARK 350 BIOMT1 10 -0.500000 -0.866025 0.000000 0.000000 REMARK 350 BIOMT2 10 0.866025 -0.500000 0.000000 0.000000 REMARK 350 BIOMT3 10 0.000000 0.000000 1.000000 -1.000000 MTRIX1 1 1.000000 0.000000 0.000000 0.00000 MTRIX2 1 0.000000 1.000000 0.000000 0.00000 MTRIX3 1 0.000000 0.000000 1.000000 0.00000 MTRIX1 1 1.000000 0.000000 0.000000 0.00000 1 MTRIX2 1 0.000000 1.000000 0.000000 0.00000 1 MTRIX3 1 0.000000 0.000000 1.000000 0.00000 1 MTRIX1 2 1.000000 0.000000 0.000000 0.00000 MTRIX2 2 0.000000 0.000000 -1.000000 0.00000 MTRIX3 2 0.000000 1.000000 0.000000 0.00000 MTRIX1 3 0.500000 -0.866025 0.000000 0.00000 MTRIX2 3 0.866025 0.500000 0.000000 0.00000 MTRIX3 3 0.000000 0.000000 1.000000 0.00000 MTRIX1 4 -0.500000 -0.866025 0.000000 0.00000 MTRIX2 4 0.866025 -0.500000 0.000000 0.00000 MTRIX3 4 0.000000 0.000000 1.000000 0.00000 MTRIX1 5 1.000000 0.000000 0.000000 0.00000 MTRIX2 5 0.000000 1.000000 0.000000 0.50000 MTRIX3 5 0.000000 0.000000 1.000000 0.00000 ATOM 1 N ILE A 40 1.000 1.000 1.000 1.00162.33 C ATOM 2 CA LEU A 40 94.618 -5.253 91.582 1.00 87.10 C ATOM 3 C ARG B 40 62.395 51.344 80.786 1.00107.25 C HETATM 4 C1 EDO A 40 39.954 51.526 72.372 0.33 60.93 C ''' pdb_test_data2="""\ MTRIX1 1 1.000000 0.000000 0.000000 0.00000 1 MTRIX2 1 0.000000 1.000000 0.000000 0.00000 1 MTRIX3 1 0.000000 0.000000 1.000000 0.00000 1 MTRIX1 2 0.496590 -0.643597 0.582393 0.00000 MTRIX2 2 0.867925 0.376088 -0.324443 0.00000 MTRIX3 2 -0.010221 0.666588 0.745356 0.00000 MTRIX1 3 -0.317946 -0.173437 0.932111 0.00000 MTRIX2 3 0.760735 -0.633422 0.141629 0.00000 MTRIX3 3 0.565855 0.754120 0.333333 0.00000 ATOM 1 N THR A 1 9.670 10.289 11.135 1.00 20.00 N ATOM 2 CA THR A 2 9.559 8.931 10.615 1.00 20.00 C ATOM 3 C THR A 3 9.634 7.903 11.739 1.00 20.00 C ATOM 4 O THR B 4 10.449 8.027 12.653 1.00 20.00 O ATOM 5 CB THR B 5 10.660 8.630 9.582 1.00 20.00 C ATOM 6 OG1 THR A 6 10.560 9.552 8.490 1.00 20.00 O ATOM 7 CG2 THR A 7 10.523 7.209 9.055 1.00 20.00 C TER """ pdb_test_data3="""\ MTRIX1 1 1.000000 0.000000 0.000000 0.00000 1 MTRIX2 1 0.000000 1.000000 0.000000 0.00000 1 MTRIX3 1 0.000000 0.000000 1.000000 0.00000 1 MTRIX1 2 0.309017 -0.809017 0.500000 0.00000 MTRIX2 2 0.809017 0.500000 0.309017 0.00000 MTRIX3 2 -0.500000 0.309017 0.809017 0.00000 MTRIX1 3 -0.809017 -0.500000 0.309017 0.00000 MTRIX2 3 0.500000 -0.309017 0.809017 0.00000 MTRIX3 3 -0.309017 0.809017 0.500000 0.00000 ATOM 749 O UNK A 90 28.392 67.262 97.682 1.00 0.00 O ATOM 750 N UNK A 91 30.420 66.924 98.358 1.00 0.00 N TER ATOM 1495 N UNK B 67 33.124 2.704 114.920 1.00 0.00 N END """ pdb_test_data4="""\ ATOM 1 N THR A 1 9.670 10.289 11.135 1.00 20.00 N ATOM 2 CA THR A 2 9.559 8.931 10.615 1.00 20.00 C ATOM 3 C THR A 3 9.634 7.903 11.739 1.00 20.00 C ATOM 4 O THR B 4 10.449 8.027 12.653 1.00 20.00 O ATOM 5 CB THR B 5 10.660 8.630 9.582 1.00 20.00 C ATOM 6 OG1 THR A 6 10.560 9.552 8.490 1.00 20.00 O ATOM 7 CG2 THR A 7 10.523 7.209 9.055 1.00 20.00 C """ pdb_test_data5="""\ MTRIX1 1 0.309017 -0.809017 0.500000 0.00000 MTRIX2 1 0.809017 0.500000 0.309017 0.00000 MTRIX3 1 -0.500000 0.309017 0.809017 0.00000 MTRIX1 2 -0.809017 -0.500000 0.309017 0.00000 MTRIX2 2 0.500000 -0.309017 0.809017 0.00000 MTRIX3 2 -0.309017 0.809017 0.500000 0.00000 MTRIX1 3 1.000000 0.000000 0.000000 0.00000 1 MTRIX2 3 0.000000 1.000000 0.000000 0.00000 1 MTRIX3 3 0.000000 0.000000 1.000000 0.00000 1 ATOM 749 O UNK A 90 28.392 67.262 97.682 1.00 0.00 O ATOM 750 N UNK A 91 30.420 66.924 98.358 1.00 0.00 N TER ATOM 1495 N UNK B 67 33.124 2.704 114.920 1.00 0.00 N END """ pdb_test_data6="""\ MTRIX1 1 0.309017 -0.809017 0.500000 0.00000 MTRIX2 1 0.809017 0.500000 0.309017 0.00000 MTRIX3 1 -0.500000 0.309017 0.809017 0.00000 MTRIX1 2 -0.809017 -0.500000 0.309017 0.00000 MTRIX2 2 0.500000 -0.309017 0.809017 0.00000 MTRIX3 2 -0.309017 0.809017 0.500000 0.00000 ATOM 749 O UNK A 90 28.392 67.262 97.682 1.00 0.00 O ATOM 750 N UNK A 91 30.420 66.924 98.358 1.00 0.00 N TER ATOM 1495 N UNK B 67 33.124 2.704 114.920 1.00 0.00 N END """ pdb_test_data3_expected_results = """\ ATOM 749 O UNK A 90 28.392 67.262 97.682 1.00 0.00 O ATOM 750 N UNK A 91 30.420 66.924 98.358 1.00 0.00 N TER ATOM 1495 N UNK B 67 33.124 2.704 114.920 1.00 0.00 N TER ATOM 749 O UNK C 90 3.199 86.786 85.616 1.00 0.00 O ATOM 750 N UNK C 91 4.437 88.467 85.044 1.00 0.00 N TER ATOM 1495 N UNK D 67 65.508 63.662 77.246 1.00 0.00 N TER ATOM 749 O UNK E 90 -26.415 72.437 94.483 1.00 0.00 O ATOM 750 N UNK E 91 -27.678 74.103 93.921 1.00 0.00 N TER ATOM 1495 N UNK F 67 7.362 108.699 49.412 1.00 0.00 N TER """ def run_selected_tests(): """ Run selected tests 1) List in "tests" the names of the particular test you want to run 2) Comment out unittest.main() 3) Un-comment unittest.TextTestRunner().run(run_selected_tests()) """ tests = ['test_spec_file_format'] suite = unittest.TestSuite(map(TestMultimerReconstruction,tests)) return suite if __name__=='__main__': # use for individual tests # unittest.TextTestRunner().run(run_selected_tests()) # Use to run all tests unittest.main()