def setUp(self) -> None:
"""Initialize the unittest."""
self.dataset = Nations()
def setUp(self):
"""Prepare for testing the evaluation structure."""
self.counter = 1337
self.evaluator = DummyEvaluator(counter=self.counter, filtered=True, automatic_memory_optimization=False)
self.triples_factory = Nations().training
self.model = MockModel(triples_factory=self.triples_factory)
def test_create_inverse_triples(self):
"""Verify that inverse triples are only created in the training factory."""
dataset = Nations(create_inverse_triples=True)
assert dataset.training.create_inverse_triples
assert not dataset.testing.create_inverse_triples
assert not dataset.validation.create_inverse_triples
def setUp(self) -> None:
"""Set up the tests."""
self.triples_factory = Nations().training
self.assertEqual(1592, self.triples_factory.num_triples)
def setUp(self) -> None:
"""Set up the test case with a triples factory."""
self.triples_factory = Nations().training
def setUp(self) -> None:
"""Instantiate triples factory and model."""
self.triples_factory = Nations().training
def setUp(self) -> None:
"""Instantiate test instance."""
self.factory = Nations().training
def setUp(self) -> None:
"""Set up the test case with a triples factory."""
self.triples_factory = Nations().training
self.num_samples = 20
self.num_epochs = 10
self.graph_sampler = GraphSampler(triples_factory=self.triples_factory, num_samples=self.num_samples)