#!/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])