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