def test_row_lpush(self): """Row lpush.""" # Arrange john = Row(self.john) george = Row(self.george) # Act john.lpush(george) # Assert self.assertEqual(john[-1], george)
def test_row_append(self): """Row append.""" # Arrange john = Row(self.john) george = Row(self.george) # Act john.append(george) # Assert self.assertEqual(john[0], george)
def test_row_has_tags(self): """Row has_tag.""" # Arrange john = Row(self.john, tags=["tag1", "tag2"]) # Act / Assert self.assertTrue(john.has_tag(["tag2", "tag1"]))
def test_row_contains(self): """Row __contains__.""" # Arrange john = Row(self.john) # Act / Assert self.assertIn("John", john)
def test_row_no_tag(self): """Row has_tag.""" # Arrange john = Row(self.john) # Act / Assert self.assertFalse(john.has_tag("not found")) self.assertFalse(john.has_tag(None))
def test_row_repr(self): """Row repr.""" # Arrange john = Row(self.john) # Act output = str(john) # Assert self.assertEqual(output, "['John', 'Adams', 90]")
def test_row_slicing(self): """Verify Row slicing. Issue #184.""" john = Row(self.john) self.assertEqual(john[:], list(self.john[:])) self.assertEqual(john[0:], list(self.john[0:])) self.assertEqual(john[:2], list(self.john[:2])) self.assertEqual(john[0:2], list(self.john[0:2])) self.assertEqual(john[0:-1], list(self.john[0:-1]))
def test_row_pickle_unpickle(self): """Row __setstate__ and __getstate__.""" # Arrange before_pickle = Row(self.john) # Act output = pickle.loads(pickle.dumps(before_pickle)) # Assert self.assertEqual(output[0], before_pickle[0]) self.assertEqual(output[1], before_pickle[1]) self.assertEqual(output[2], before_pickle[2])
def test_row_append(self): """Row append.""" john = Row(self.john) john.append('stuff') self.assertEqual(john.list, ['John', 'Adams', 90, 'stuff'])
def test_row_lpush(self): """Row lpush.""" john = Row(self.john) john.lpush(53) self.assertEqual(john.list, [53, 'John', 'Adams', 90])