Skip to content

Commit

Permalink
Update test case
Browse files Browse the repository at this point in the history
  • Loading branch information
emjun committed Aug 2, 2021
1 parent 03ba9ae commit 64c49d4
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion tests/test_graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -207,6 +207,12 @@ def test_add_interaction_effects(self):
design = ts.Design(dv=test_score, ivs=[race, ses])

gr = design.graph
self.assertTrue(gr.has_variable(student))
self.assertTrue(gr.has_variable(race))
self.assertTrue(gr.has_variable(ses))
self.assertTrue(gr.has_variable(test_score))
self.assertFalse(gr.has_variable(tutoring))

identifiers = gr.get_identifiers()
self.assertIn(student, identifiers)
self.assertTrue(gr.has_edge(student, race, "has"))
Expand Down Expand Up @@ -243,7 +249,7 @@ def test_add_interaction_effects(self):
self.assertEqual(count_has_edges, 4) # Should this be 5?
self.assertEqual(count_associates_edges, 6)
self.assertEqual(count_other_edges, 0)

# Identifier has interaction effect only once (no duplicates)
# Associate (between Race*SES and Test score introduces to edges)
self.assertEqual(len(edges), 10)
Expand Down

0 comments on commit 64c49d4

Please sign in to comment.