def test_build_graph(): file = open(os.path.abspath(os.path.join(os.path.dirname(__file__), "kmer.pck")),'rb') kmer_dict = pickle.load(file) graph = build_graph(kmer_dict) #TCAGAGA #TCA TC CA #CAG CA AG #AGA AG GA #GAG GA AG #AGA AG GA assert graph.number_of_nodes() == 4 assert graph.number_of_edges() == 4 assert "AG" in graph assert "GA" in graph assert graph.edges["AG", "GA"]['weight'] == 2 file.close() # def test_build_graph_comp(): # file = open(os.path.abspath(os.path.join(os.path.dirname(__file__), "kmer_comp.pck")),'rb') # kmer_dict = pickle.load(file) # graph = build_graph(kmer_dict) # #TCAGAGA # #TCA TC CA # #CAG CA AG # #AGA AG GA # #GAG GA AG # #AGA AG GA # # ((TC, AG), (CA, GT)), (CA, AG), (AG # assert graph.number_of_nodes() == 4 # assert graph.number_of_edges() == 3 # assert "AG" in graph # assert "GA" in graph # assert graph.edges["AG", "GA"]['weight'] == 2
def test_build_graph(): file = open( os.path.abspath(os.path.join(os.path.dirname(__file__), "kmer.pck")), 'rb') kmer_dict = pickle.load(file) graph = build_graph(kmer_dict) #TCAGAGA #TCA TC CA #CAG CA AG #AGA AG GA #GAG GA AG #AGA AG GA assert graph.number_of_nodes() == 4 assert graph.number_of_edges() == 4 assert "AG" in graph assert "GA" in graph assert graph.edges["AG", "GA"]['weight'] == 2