chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,40 @@
|
||||
import unittest
|
||||
|
||||
import easygraph as eg
|
||||
import numpy as np
|
||||
|
||||
|
||||
class TestHypergraphSCentrality(unittest.TestCase):
|
||||
def setUp(self):
|
||||
# Simple test hypergraph
|
||||
self.hg = eg.Hypergraph(num_v=5, e_list=[(0, 1), (1, 2, 3), (3, 4)])
|
||||
self.empty_hg = eg.Hypergraph(num_v=1, e_list=[])
|
||||
self.singleton_hg = eg.Hypergraph(num_v=3, e_list=[(0,), (1,), (2,)])
|
||||
|
||||
def test_s_betweenness_normal(self):
|
||||
result = eg.s_betweenness(self.hg)
|
||||
self.assertIsInstance(result, (list, dict))
|
||||
self.assertTrue(all(isinstance(x, (int, float)) for x in result))
|
||||
|
||||
def test_s_closeness_normal(self):
|
||||
result = eg.s_closeness(self.hg)
|
||||
self.assertIsInstance(result, (list, dict))
|
||||
self.assertTrue(all(isinstance(x, (int, float)) for x in result))
|
||||
|
||||
def test_s_eccentricity_all(self):
|
||||
result = eg.s_eccentricity(self.hg)
|
||||
self.assertIsInstance(result, dict)
|
||||
for v in result.values():
|
||||
self.assertIsInstance(v, (int, float, np.integer, np.floating))
|
||||
|
||||
def test_s_eccentricity_edges_false(self):
|
||||
result = eg.s_eccentricity(self.hg, edges=False)
|
||||
self.assertIsInstance(result, dict)
|
||||
|
||||
def test_s_eccentricity_invalid_source(self):
|
||||
with self.assertRaises(KeyError):
|
||||
eg.s_eccentricity(self.hg, source=(999, 888))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user