def test_iteration(self): cs = CaseArray() cs.record(self.case1) cs.record(self.case2) cs.record(self.case1_dup) expected = [self.case1, self.case2, self.case1_dup] for i,case in enumerate(cs): self.assertEqual(case._inputs, expected[i]._inputs) self.assertEqual(case._outputs, expected[i]._outputs)
def test_from_case(self): cs = CaseArray(self.case1) self.assertEqual(1, len(cs)) self.assertEqual(cs[0]._inputs, self.case1_dup._inputs) self.assertEqual(cs[0]._outputs, self.case1_dup._outputs) cs.record(self.case2) cs.record(self.case1_dup) self.assertEqual(3, len(cs)) self.assertTrue(self.case2 in cs)
def test_start_empty(self): cs = CaseArray() cs.record(self.case1) cs.record(self.case2) cs.record(self.case1_dup) self.assertEqual(3, len(cs)) self.assertEqual(cs[0]._inputs, self.case1_dup._inputs) self.assertEqual(cs[0]._outputs, self.case1_dup._outputs) self.assertEqual(cs[1]._inputs, self.case2._inputs) self.assertEqual(cs[1]._outputs, self.case2._outputs) self.assertEqual(cs[2]._inputs, self.case1_dup._inputs) self.assertEqual(cs[2]._outputs, self.case1_dup._outputs)
def test_copy(self): cs = CaseArray() cs.record(self.case1) cs.record(self.case2) cs.record(self.case1_dup) cscopy = cs.copy() for c1, c2 in zip(cs, cscopy): self.assertEqual(c1, c2)
def test_from_dict(self): dct = { 'comp1.a': [2, 4, 6], 'comp1.b': [4, 8, 12], 'comp2.b': [1, 2, 3], } cs = CaseArray(dct) self.assertEqual(3, len(cs)) self.assertEqual(set(['comp1.a', 'comp1.b', 'comp2.b']), set(cs._names)) self.assertEqual(cs['comp1.a'], [2, 4, 6]) case = cs[1] expected = Case(inputs=[('comp1.a', 4), ('comp1.b', 8), ('comp2.b', 2)]) self.assertEqual(case._inputs, expected._inputs)
def test_iteration(self): cs = CaseArray() cs.record(self.case1) cs.record(self.case2) cs.record(self.case1_dup) expected = [self.case1, self.case2, self.case1_dup] for i, case in enumerate(cs): self.assertEqual(case._inputs, expected[i]._inputs) self.assertEqual(case._outputs, expected[i]._outputs)
def test_contains(self): ca = CaseArray() ca.record(self.case1) self.assertTrue(self.case1_dup in ca) self.assertFalse(self.case2 in ca) self.assertFalse(None in ca)