#!/usr/bin/env python
from nose.tools import *
import networkx as nx

class TestIntersectionGraph():
    def test_random_intersection_graph(self):
        G=nx.uniform_random_intersection_graph(10,5,0.5)
        assert_equal(len(G),10)

    def test_k_random_intersection_graph(self):
        G=nx.k_random_intersection_graph(10,5,2)
        assert_equal(len(G),10)

    def test_general_random_intersection_graph(self):
        G=nx.general_random_intersection_graph(10,5,[0.1,0.2,0.2,0.1,0.1])
        assert_equal(len(G),10)
        assert_raises(ValueError, nx.general_random_intersection_graph,10,5,
                      [0.1,0.2,0.2,0.1])